Nextjournal's PyTorch environment runs PyTorch v1.3.1, and is configured to use Nvidia CUDA v10.2. ... correct += pred.eq(target.view_as(pred)).sum().item()
確定! 回上一頁