YWV
Results
1
comments of
YWV
add one = one.repeat(32,1,1,1) mone = mone.repeat(32,1,1,1) and set drop_last=True in dataloader should fix this error.