Add prelu to list of torch overrides (#217)
* Add prelu to list of torch overrides
This is to fix the following error:
File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/container.py", line 92, in forward
input = module(input)
File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in __call__
result = self.forward(*input, **kwargs)
File "/opt/conda/lib/python3.6/site-packages/torch/nn/modules/activation.py", line 722, in forward
return F.prelu(input, self.weight)
File "/opt/conda/lib/python3.6/site-packages/torch/nn/functional.py", line 1040, in prelu
return torch.prelu(input, weight)
RuntimeError: expected scalar type Half but found Float
* Update torch_overrides.py
Showing
Please register or sign in to comment