Approx 3:2 biased towards negative samples. What about you? I’m thinking about trying with torch.utils.data.sampler.WeightedRandomSampler to counteract the dataset imbalance problem. However, I’m not sure dice coefficient is the most suitable evaluation metric (it was not specified in the paper).