Using a target size (torch.Size([16, 1, 64, 64])) that is different to the input size (torch.Size([15, 1, 64, 64])) is deprecated. Please ensure they have the same size.

TobiasSunderdiek / cartoon-gan

Implementation of cartoon GAN [Chen et al., CVPR18] with pytorch

https://tobiassunderdiek.github.io/cartoon-gan/

MIT License

61 stars 22 forks source link

Using a target size (torch.Size([16, 1, 64, 64])) that is different to the input size (torch.Size([15, 1, 64, 64])) is deprecated. Please ensure they have the same size. #6

Closed bdlneto closed 2 years ago

bdlneto commented 2 years ago

Hi again, @TobiasSunderdiek

During the trainning process, this error raised and I have no Idea of what is causing it. I am using the default dataset (safebooru) and default params for the trainning.

Do you have any idea on how to fix it?

Regards

TobiasSunderdiek commented 2 years ago

Hi @bdlneto,

thanks again for the issue, every issue makes this repo better!

I think this is caused by processing a batch smaller than 16. Does this error occur while processing the training cell in the notebook?

At some places in the code the actual_batch_size is used to avoid this, but it seems I missed another place.

I think during the download of the images, some error occured. Can you please check if all images in the cartoons/, cartoons_smoothed/ and photos/ folder contain the same amount of 4000 images?

Thanks and best regards!

bdlneto commented 2 years ago

Hi!

I also found it to be some problem during the download process.

I'll check the folders and, if necessary, re-download them

Thanks again!

TobiasSunderdiek commented 2 years ago

Great, thank you for helping me making this repo better by having opened this issue!

I'll close it now, best regards