search

hhb072 / WaveletSRNet

A pytorch implementation of Paper "Wavelet-srnet: A wavelet-based cnn for multi-scale face super resolution"
MIT License
240 stars 67 forks source link

The size of tensor a (128) must match the size of tensor b (32) at non-singleton dimension 3 #19

Open dundee98 opened 11 months ago

dundee98 commented 11 months ago

could anyone else help me ,i fail to run this code . it shows in this part of code: loss_lr = loss_MSE(wavelets_predict[:,0:3,:,:], wavelets_lr, opt.mse_avg) loss_sr = loss_MSE(wavelets_predict[:, 3:, :, :], wavelets_sr, opt.mse_avg) loss_textures = loss_Textures(wavelets_predict[:, 3:, :, :], wavelets_sr) loss_img = loss_MSE(img_predict, target, opt.mse_avg) the "wavelets_predict[:,0:3,:,:]" tensor.size ([16, 48, 128, 128]), but "wavelets_lr" is([16, 3, 32, 32]), i wanna know did anyone meet the same problem, how did you solve it?