PyTorch Forums
Will moving tensor from GPU to CPU to GPU break back-propagation?
vision
albanD
(Alban D)
September 24, 2020, 6:24pm
2
Hi,
No moving Tensors across devices is perfectly differentiable. It will all work fine!
3 Likes
show post in topic