Training stuck at the end of the first epoch

milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

GNU General Public License v3.0

9.3k stars 2.51k forks source link

Training stuck at the end of the first epoch #504

Closed XxUpUp closed 4 months ago

XxUpUp commented 4 months ago

Hi, I'm having the problem that when the last batch of my first epoch is trained, the program can't start the next epoch but stays stuck there and no error message appears, what could be the reason for this? (Note: I have made some changes to train.py, but not the main body of the code)