Skip to content
This repository has been archived by the owner on Mar 21, 2024. It is now read-only.

Fix for SDK regression bug, minor doc updates #475

Merged
merged 19 commits into from
Jun 4, 2021
Prev Previous commit
Next Next commit
adding test
  • Loading branch information
ant0nsc committed Jun 4, 2021
commit 614557339ed76ba53f2d0b7b0d3dfb9a7a7fec2e
4 changes: 1 addition & 3 deletions InnerEye/ML/configs/segmentation/Lung.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,9 +60,7 @@ def __init__(self, **kwargs: Any) -> None:
use_mixed_precision=True,
use_model_parallel=True,
monitoring_interval_seconds=0,
loss_type=SegmentationLoss.Mixture,
mixture_loss_components=[MixtureLossComponent(0.5, SegmentationLoss.Focal, 0.2),
MixtureLossComponent(0.5, SegmentationLoss.SoftDice, 0.1)],
loss_type=SegmentationLoss.SoftDice,
check_exclusive=False,
)
self.add_and_validate(kwargs)
Expand Down
2 changes: 1 addition & 1 deletion azure-pipelines/build-pr.yml
Original file line number Diff line number Diff line change
Expand Up @@ -203,7 +203,7 @@ jobs:
- name: tag
value: 'LungPR'
- name: more_switches
value: '--restrict_subjects=8,8,1 --num_epochs=1'
value: '--restrict_subjects=4,4,1 --num_epochs=1 --train_batch_size=4'
pool:
vmImage: 'ubuntu-18.04'
steps:
Expand Down