search

knazeri / edge-informed-sisr

Edge-Informed Single Image Super-Resolution, ICCVW 2019 https://arxiv.org/abs/1909.05305
Other
77 stars 20 forks source link

TypeError: torch.FloatTensor is not a Module subclass #1

Open qingfengmingyue opened 5 years ago

qingfengmingyue commented 5 years ago

Traceback (most recent call last): File "train.py", line 2, in main(mode=1) File "/root/edge-informed-sisr/main.py", line 42, in main model = EdgeMatch(config) File "/root/edge-informed-sisr/src/edge_match.py", line 27, in init self.sr_model = SRModel(config).to(config.DEVICE) File "/root/edge-informed-sisr/src/models.py", line 179, in init self.add_module('scale_kernel', kernel_weight) File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 179, in add_module torch.typename(module))) TypeError: torch.FloatTensor is not a Module subclass

Dayananda-V commented 4 years ago

@qingfengmingyue

i found the working fix, please add the below code will fix this problem(just comment the add_module for scale_kernel instead create scale_kernel variable).

#self.add_module('scale_kernel', kernel_weight) self.scale_kernel = torch.tensor(np.tile(kernel, (3, 1, 1, 1))).float()

jsjs0827 commented 4 years ago

@Dayananda-V Can you show the result? I just figure out the mosaic.