如果需要从gpu 转到cpu 的话,PyTorch 会自动帮你处理。 ... This in only valid if loss is a tensor containing a single element.
確定! 回上一頁