Skip to content
Snippets Groups Projects

Replace sampler balancing by loss balancing

Merged Daniel CARRON requested to merge loss-balancing into main
Files
10
@@ -15,8 +15,7 @@ from mednet.data.augmentations import ElasticDeformation
from mednet.models.alexnet import Alexnet
model = Alexnet(
train_loss_type=BCEWithLogitsLoss,
validation_loss_type=BCEWithLogitsLoss,
loss_type=BCEWithLogitsLoss,
optimizer_type=SGD,
optimizer_arguments=dict(lr=0.01, momentum=0.1),
augmentation_transforms=[ElasticDeformation(p=0.8)],
Loading