search

BingyaoHuang / CompenNet

[CVPR'19] End-to-end Projector Photometric Compensation
Other
36 stars 18 forks source link

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same #2

Closed sapoluri closed 4 years ago

sapoluri commented 4 years ago

The execution bails out at the end of the first iteration with the above error:

How to fix it: everywhere you call .to(device) you need to change it to = .to(device).

The "to" function call isn't modifying the object it's called upon.

BingyaoHuang commented 4 years ago

Thank you. Fixed in 76f2896.