search

KeepTryingTo / Pytorch-GAN

使用Pytorch实现GAN 的过程
52 stars 5 forks source link

Wgangp Grayscale images #1

Open Cherish420 opened 1 year ago

Cherish420 commented 1 year ago 
#Hello author, I have some questions about wgangp. If I use grayscale images as input and change the number of channels to 1, the code will result in the following error：RuntimeError: Given groups=1, weight of size [16, 1, 4, 4], expected input[16, 3, 28, 28] to have 1 channels, but got 3 channels instead
KeepTryingTo commented 1 year ago

     when you have changed the channels,you should also pay attention the generator and discriminator's model.

------------------ 原始邮件 ------------------ 发件人: "KeepTryingTo/Pytorch-GAN" @.>; 发送时间: 2023年6月20日(星期二) 晚上10:37 @.>; @.***>; 主题: [KeepTryingTo/Pytorch-GAN] Wgangp Grayscale images (Issue #1)

Hello author, I have some questions about wgangp. If I use grayscale images as input and change the number of channels to 1, the code will result in the following error：RuntimeError: Given groups=1, weight of size [16, 1, 4, 4], expected input[16, 3, 28, 28] to have 1 channels, but got 3 channels instead

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>