jeong-tae / RACNN-pytorch

This is a third party implementation of RA-CNN in pytorch.
201 stars 63 forks source link

RuntimeError: _th_exp_out not supported on CUDAType for Long #18

Closed miss1997yuan closed 4 years ago

miss1997yuan commented 4 years ago

is there someone meet the same problem ?

Pythonisfun commented 4 years ago

i have the same problem,have you solved it?

pshroff04 commented 4 years ago

May be little late here. Line 75 in RACNN.py:
h = lambda x: 1 / (1 + torch.exp(-10 * x)) to h = lambda x: 1 / (1 + torch.exp(-10. * x)) (10.)