INFO: Using device cuda
INFO: Network:
NestedUNet model
3 input channels
3 output channels (classes)
Bilinear upscaling
INFO: Creating dataset with 20 examples
training stops after this. can anyone help