pytc
/

mito / MitoEM-R-BC.yaml
dwei's picture
Upload 2 files
7a47fa0 verified
raw
history blame contribute delete
439 Bytes
MODEL:
OUT_PLANES: 2
TARGET_OPT: ["0", "4-1-1"]
LOSS_OPTION:
- - WeightedBCEWithLogitsLoss
- DiceLoss
- - WeightedBCEWithLogitsLoss
- DiceLoss
LOSS_WEIGHT: [[1.0, 0.5], [1.0, 0.5]]
WEIGHT_OPT: [["1", "0"], ["1", "0"]]
OUTPUT_ACT: [["none", "sigmoid"], ["none", "sigmoid"]]
INFERENCE:
OUTPUT_ACT: ["sigmoid", "sigmoid"]
OUTPUT_PATH: outputs/MitoEM_R_BC/test/
DATASET:
OUTPUT_PATH: outputs/MitoEM_R_BC/