Please
- Update to the latest docker which is released yesterday. See Migrating to TAO Toolkit — TAO Toolkit 3.0 documentation and NVIDIA NGC
- Please use
loss: "cross_dice_sum"
- Please convert the mask images to gray images. After checking the public dataset you mentioned, the pixel value is either 0 or 128. Please map 128 to 1.
- For binary segmentation, the label_id should be 0 and 1. BTW, if there are 4 classes , label_id should belong to
0-3