![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
机器学习
「已注销」
顺其自然
展开
-
PyTorch学习笔记(一)
PyTorch学习笔记(一)PyTorch学习笔记PyTorch中的数据类型自动求梯度未完待续……来自 https://pytorch.org/tutorials/beginner/deep_learning_60min_blitz.htmlPyTorch学习笔记类似TensorFlow,PyTorch也是一个深度学习的框架,另外它也作为NumPy中的数据在GPU上运算时使用的一种替代品...原创 2019-02-27 20:27:07 · 301 阅读 · 0 评论 -
pytorch使用GPU训练MNIST数据集
参考莫凡博客进行MNIST数据集的训练,临时记录所使用的代码。import torchimport torch.nn as nnimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as plttorch.manual_seed(1)EPOCH = 1BATCH_SIZE = 50...原创 2019-02-24 12:44:13 · 4311 阅读 · 0 评论