RuntimeError: Caught RuntimeError in DataLoader worker process 0. && RuntimeError: Trying to resize storage that is not resizable

ykdai / BasicPBC

Official Implementation of "Learning Inclusion Matching for Animation Paint Bucket Colorization"

Other

233 stars 23 forks source link

Closed shawnl111 closed 5 months ago

shawnl111 commented 5 months ago

training on torch2.2.1 and cuda11.8 on widows11

riderexin commented 2 months ago

have the same problem. if num_worker>0 and batch_size>1, I have the same error. if num_worker = 0 and batch_size>1, stack error.

num_worker = 0 and batch_size=1, it is ok

riderexin commented 2 months ago

The dataset itself has tensors of different shapes. So they can't be stacked anyway, in my opinion.