pytorch_ssim module

[YoungJoongUNC]

Hello.

In your loss.py, you import pytorch_ssim. So I installed the pytorch package using pip. But I got following error during the training. Did you use other pytorch_ssim package in your training?

  File "/playpen/youngjoong/code/TDAN-VSR/solver.py", line 420, in train
    train_psnr, train_ssim, _, _, _ = self._check_PSNR(val_dataset)
  File "/playpen/youngjoong/code/TDAN-VSR/solver.py", line 333, in _check_PSNR
    ssim = pytorch_ssim.ssim(output_batch + 0.5, label_batch + 0.5, size_average=False)
  File "/home/youngjoong/anaconda3/envs/deform/lib/python3.6/site-packages/pytorch_ssim/__init__.py", line 62, in ssim
    return _ssim(img1, img2, window, window_size, channel, size_average)
  File "/home/youngjoong/anaconda3/envs/deform/lib/python3.6/site-packages/pytorch_ssim/__init__.py", line 18, in _ssim
    mu1 = F.conv2d(img1, window, padding = window_size/2, groups = channel)
  File "/home/youngjoong/anaconda3/envs/deform/lib/python3.6/site-packages/torch/nn/functional.py", line 89, in conv2d
    torch.backends.cudnn.deterministic, torch.backends.cudnn.enabled)
RuntimeError: argument 1 (padding) must be tuple of int but got tuple of (float, float)

[YapengTian]

Yes, it has been uploaded.

[YoungJoongUNC]

Thank you!