vdigpku / qgcn Goto Github PK

(qgcn) rong@server02:/sda/qgcn/QGCN-master$ python src/eval_qmg.py --model /sda/qgcn/QGCN-master/QCN_RGB.pth --GT --gtdata DIV2K --quality 10 --cuda
Namespace(GT=True, colorMode='RGB', cuda=True, gpus='0', gtdata='DIV2K', model='/sda/qgcn/QGCN-master/QCN_RGB.pth', quality=10, realdatapath='testdata', save='')
=> use gpu id: '0'
/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/serialization.py:868: SourceChangeWarning: source code of class 'torch.nn.modules.conv.Conv2d' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
  warnings.warn(msg, SourceChangeWarning)
/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/serialization.py:868: SourceChangeWarning: source code of class 'torch.nn.modules.container.Sequential' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
  warnings.warn(msg, SourceChangeWarning)
/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/serialization.py:868: SourceChangeWarning: source code of class 'torch.nn.modules.activation.ReLU' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
  warnings.warn(msg, SourceChangeWarning)
/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/serialization.py:868: SourceChangeWarning: source code of class 'torch.nn.modules.pixelshuffle.PixelShuffle' has changed. you can retrieve the original source code by accessing the object's source attribute or set `torch.nn.Module.dump_patches = True` and use the patch tool to revert the changes.
  warnings.warn(msg, SourceChangeWarning)

[1/100] 0801.png
Traceback (most recent call last):
  File "src/eval_qmg.py", line 101, in <module>
    HQImg = model([currImg, gloImg])
  File "/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
    return forward_call(*input, **kwargs)
  File "/sda/qgcn/QGCN-master/src/nets/qmar.py", line 39, in forward
    out = self.conv_input(x)
  File "/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
    return forward_call(*input, **kwargs)
  File "/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 463, in forward
    return self._conv_forward(input, self.weight, self.bias)
  File "/home/rong/anaconda3/envs/qgcn/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 459, in _conv_forward
    return F.conv2d(input, weight, bias, self.stride,
TypeError: conv2d() received an invalid combination of arguments - got (list, Parameter, NoneType, tuple, tuple, tuple, int), but expected one of:
 * (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, tuple of ints padding, tuple of ints dilation, int groups)
      didn't match because some of the arguments have invalid types: (list, Parameter, NoneType, tuple, tuple, tuple, int)
 * (Tensor input, Tensor weight, Tensor bias, tuple of ints stride, str padding, tuple of ints dilation, int groups)
      didn't match because some of the arguments have invalid types: (list, Parameter, NoneType, tuple, tuple, tuple, int)