Open MichaelMonashev opened 5 years ago
Same issue, you could try adding drop_last=True in Dataloader in sample_data function in train.py
I added drop_last=True
in Dataloader, but got this error again.
Sometimes this error is raised on 128 resolution, sometimes not.
Maybe some images in your dataset have crashed?