1
发布时间 2023-04-08 21:58:10作者: helloWorldhelloWorld
File "/media/mmsys/6f1091c9-4ed8-4a10-a03d-2acef144d2e1/SXY/FLW-Net-main/lowlight_train.py", line 164, in <module>
train(config)
File "/media/mmsys/6f1091c9-4ed8-4a10-a03d-2acef144d2e1/SXY/FLW-Net-main/lowlight_train.py", line 89, in train
loss_2, Loss_ssim = L_recon(enhanced_image,img_highlight)
File "/home/mmsys/anaconda3/envs/FLW/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/media/mmsys/6f1091c9-4ed8-4a10-a03d-2acef144d2e1/SXY/FLW-Net-main/Myloss.py", line 33, in forward
L2 = (1- self.ssim_loss(R_low,high)).mean()
File "/home/mmsys/anaconda3/envs/FLW/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/mmsys/anaconda3/envs/FLW/lib/python3.8/site-packages/pytorch_ssim/__init__.py", line 57, in forward
return _ssim(img1, img2, window, self.window_size, channel, self.size_average)
File "/home/mmsys/anaconda3/envs/FLW/lib/python3.8/site-packages/pytorch_ssim/__init__.py", line 18, in _ssim
mu1 = F.conv2d(img1, window, padding = window_size/2, groups = channel)
TypeError: conv2d() received an invalid combination of arguments - got (Tensor, Tensor, groups=int, padding=float), but expected one of:
* (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, tuple of ints padding, tuple of ints dilation, int groups)
* (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, str padding, tuple of ints dilation, int groups)