trianand's picture
Update model.yaml
31ca68a verified
raw
history blame
220 Bytes
$autrainer.models.cnn_10.Cnn10==0.3.0:
output_dim: 4
sigmoid_output: false
sigmoid_predictions: true
segmentwise: false
in_channels: 1
transfer: https://zenodo.org/records/3987831/files/Cnn10_mAP%3D0.380.pth