That’s quite strange, as I’m using just the code you’ve posted here.
These are my results:
Epoch 0, loss 1.2339667081832886, bce 0.7199727296829224, dice 0.5139939785003662
Epoch 1, loss 1.220333218574524, bce 0.7162541151046753, dice 0.5040791034698486
Epoch 2, loss 1.2090554237365723, bce 0.7133639454841614, dice 0.49569153785705566
Epoch 3, loss 1.197749137878418, bce 0.7105355262756348, dice 0.487213671207428
Epoch 4, loss 1.1889395713806152, bce 0.7084173560142517, dice 0.4805222153663635
Epoch 5, loss 1.1799421310424805, bce 0.7061132192611694, dice 0.4738289713859558
Epoch 6, loss 1.1723219156265259, bce 0.7041332125663757, dice 0.46818870306015015
Epoch 7, loss 1.1651051044464111, bce 0.7022958993911743, dice 0.4628092050552368
Epoch 8, loss 1.1579382419586182, bce 0.7005487084388733, dice 0.4573895335197449
Epoch 9, loss 1.1515626907348633, bce 0.6988978385925293, dice 0.45266491174697876
Epoch 10, loss 1.145705223083496, bce 0.697318434715271, dice 0.4483868479728699
Epoch 11, loss 1.1399022340774536, bce 0.6957539319992065, dice 0.44414830207824707
Epoch 12, loss 1.1344850063323975, bce 0.6943031549453735, dice 0.4401818513870239
Epoch 13, loss 1.128913402557373, bce 0.6928091645240784, dice 0.4361041784286499
Epoch 14, loss 1.1239606142044067, bce 0.6913862824440002, dice 0.4325743317604065
Epoch 15, loss 1.1189703941345215, bce 0.6899511814117432, dice 0.42901915311813354
Epoch 16, loss 1.1143808364868164, bce 0.6886655688285828, dice 0.4257153272628784
Epoch 17, loss 1.110227108001709, bce 0.6875361800193787, dice 0.42269086837768555
Epoch 18, loss 1.1056132316589355, bce 0.6861843466758728, dice 0.4194289445877075
Epoch 19, loss 1.1010327339172363, bce 0.6848099231719971, dice 0.41622287034988403

Are you sure you’re using exactly this code you’ve shared?