Tensor 0 device cuda 0 (20/31)

Tensor 0 device cuda 0