Skip to content
Snippets Groups Projects

[densenet] adding drop_rate in model params

Merged Oscar JIMENEZ requested to merge dropout into main
4 files
+ 10
4
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -19,6 +19,7 @@ model = Densenet(
validation_loss=BCEWithLogitsLoss(),
optimizer_type=Adam,
optimizer_arguments=dict(lr=0.0001),
augmentation_transforms=[ElasticDeformation(p=0.8)],
augmentation_transforms=[ElasticDeformation(p=0.2)],
pretrained=False,
dropout=0.1,
)
Loading