tracklab
tracklab copied to clipboard
Correctly transfer tensors to device for transformers_api
The transformers api didn't work on GPU, because the tensors weren't sent to the correct device.