search

sergeytulyakov / mocogan

MoCoGAN: Decomposing Motion and Content for Video Generation
573 stars 113 forks source link

ValueError: Expected target size (16, 16, 256, 256), got torch.Size([16]) #31

Open Fanny-Yuan opened 3 years ago

Fanny-Yuan commented 3 years ago

Hi,I'm training my model and I set the batch_size, image_batch, video_batch all equal to 16 but I met that problem. The problem occurs at: File "/home/ydj/MoCoGAN/trainers.py", line 268, in train self.video_batch_size, use_categories=self.use_categories) File "/home/ydj/MoCoGAN/trainers.py", line 180, in train_discriminator l_discriminator += self.category_criterion(real_categorical.squeeze(), categories_gt.long()) Is anyone know how to solve that?