trianand commited on
Commit
31ca68a
·
verified ·
1 Parent(s): 822e1b5

Update model.yaml

Browse files

As this model should output probabilities, rather than raw logits, I set the value to true

Files changed (1) hide show
  1. model.yaml +7 -7
model.yaml CHANGED
@@ -1,7 +1,7 @@
1
- $autrainer.models.cnn_10.Cnn10==0.3.0:
2
- output_dim: 4
3
- sigmoid_output: false
4
- sigmoid_predictions: false
5
- segmentwise: false
6
- in_channels: 1
7
- transfer: https://zenodo.org/records/3987831/files/Cnn10_mAP%3D0.380.pth
 
1
+ $autrainer.models.cnn_10.Cnn10==0.3.0:
2
+ output_dim: 4
3
+ sigmoid_output: false
4
+ sigmoid_predictions: true
5
+ segmentwise: false
6
+ in_channels: 1
7
+ transfer: https://zenodo.org/records/3987831/files/Cnn10_mAP%3D0.380.pth