fix: name the output group after the module the model exposes
Browse filesoutputs_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.
- 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 |
-
|
| 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:
|