Tensor 0 device cuda 0 (2/31)

Tensor 0 device cuda 0