Andrés Piñeiro Martín

icon company University of Vigo / Balidea S.L. icon location Vigo icon location Industrial PhD candidate at the University of Vigo and the company Balidea. Interested in SLT and AI in general.

Results 1 comments of Andrés Piñeiro Martín

RuntimeError: expected scalar type Half but found Float

Hi! I've solved by adding: ``` with torch.autocast("cuda"): trainer.train() ```