model.yaml

$autrainer.models.cnn_10.Cnn10==0.3.0:
  output_dim: 4
  sigmoid_output: false
  sigmoid_predictions: false
  segmentwise: false
  in_channels: 1
  transfer: https://zenodo.org/records/3987831/files/Cnn10_mAP%3D0.380.pth