VBoussot commited on
Commit
db04a8e
·
verified ·
1 Parent(s): 15f7c94

fix: name the output group after the module the model exposes

Browse files

outputs_dataset keyed on Head:Tanh, a path of the former Python model; the config builds the model from UNetPlusPlus.yml, which exposes Tanh, so prediction failed to resolve the output group.

Files changed (1) hide show
  1. Finetune/Prediction.yml +1 -1
Finetune/Prediction.yml CHANGED
@@ -89,7 +89,7 @@ Predictor:
89
  use_cache: false
90
  batch_size: 8
91
  outputs_dataset:
92
- Head:Tanh:
93
  OutputDataset:
94
  name_class: OutSameAsGroupDataset
95
  before_reduction_transforms:
 
89
  use_cache: false
90
  batch_size: 8
91
  outputs_dataset:
92
+ Tanh:
93
  OutputDataset:
94
  name_class: OutSameAsGroupDataset
95
  before_reduction_transforms: