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

[Fanny-Yuan]

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?