PyTorch 使用笔记

因为最近在使PyTorch，在这记录一下PyTorch使用的技巧.

CUDA

在使用显卡时候需要用CUDA，model、和输入要CUDA。 注意：

直接TensorName.cuda()是不行的要 TensorName = TensorName.cuda()