Skip to content

Correctly pass tensor to device

Yoann Schneiderrequested to merge
correctly-pass-tensor-to-gpu into main

According to the docs, input_tensor.to(device) doesn't move the tensor to the device but returns a tensor that's on the device.

Fixes GPU predictions

Merge request reports