Sorry to bother. I tried to run the code you provided, but got raise the batch.exc_type(batch.exc_msg) error after hundreds of batches, note that photos have been resized to [256,256], and drop_last=True, I don't know other related factors to fix this.
Sorry to bother. I tried to run the code you provided, but got raise the batch.exc_type(batch.exc_msg) error after hundreds of batches, note that photos have been resized to [256,256], and drop_last=True, I don't know other related factors to fix this.