Update model.yaml
Browse filesAs this model should output probabilities, rather than raw logits, I set the value to true
- 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:
|
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
|