TypeError: torch.FloatTensor is not a Module subclass

[qingfengmingyue]

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]

@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]

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