pytorch
gstc123
这个作者很懒,什么都没留下…
展开
-
pytorch 与numpy
https://pytorch.org/tutorials/beginner/blitz/tensor_tutorial.html#sphx-glr-beginner-blitz-tensor-tutorial-py笔记The Torch Tensor and NumPy array will share their underlying memory locations (if the ...原创 2019-12-17 18:09:48 · 157 阅读 · 0 评论 -
pytorch GPU 安装 无10.2对应版本
咋办,,怕有问题,,原创 2019-12-16 16:49:27 · 4484 阅读 · 7 评论 -
利用随机数种子来使pytorch中的结果可以复现
https://cloud.tencent.com/developer/article/1149041如果没有涉及其他随机处理的话这两行可以固定pytorch中的随机数。SEED = 0torch.manual_seed(SEED)torch.cuda.manual_seed(SEED)...原创 2019-12-13 13:11:06 · 210 阅读 · 0 评论 -
Model.zero_grad() or optimizer.zero_grad()?
https://discuss.pytorch.org/t/model-zero-grad-or-optimizer-zero-grad/28426model.zero_grad() and optimizer.zero_grad() are the same IF all your model parameters are in that optimizer. I found it is s...原创 2019-12-13 10:14:15 · 214 阅读 · 0 评论