Traceback (most recent call last):
File "main/train.py", line 226, in
train(dataset, parser)
File "main/train.py", line 132, in train
out2, out3, out4, out5, pose = net(image)
File "/usr/local/anaconda3/envs/u2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "/home/senlinren/LR/ICON-main/main/model/icon/icon.py", line 221, in forward
x1 = self.dfa1(x1)
File "/usr/local/anaconda3/envs/u2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "/home/senlinren/LR/ICON-main/main/model/icon/modules.py", line 134, in forward
p3 = self.atrConv(f)
File "/usr/local/anaconda3/envs/u2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "/usr/local/anaconda3/envs/u2/lib/python3.7/site-packages/torch/nn/modules/container.py", line 204, in forward
input = module(input)
File "/usr/local/anaconda3/envs/u2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "/usr/local/anaconda3/envs/u2/lib/python3.7/site-packages/torch/nn/modules/activation.py", line 1240, in forward
return F.prelu(input, self.weight)
RuntimeError: expected scalar type Half but found Float