vbals的博客

PyTorch学习笔记（一）PyTorch学习笔记PyTorch中的数据类型自动求梯度未完待续……来自 https://pytorch.org/tutorials/beginner/deep_learning_60min_blitz.htmlPyTorch学习笔记类似TensorFlow，PyTorch也是一个深度学习的框架，另外它也作为NumPy中的数据在GPU上运算时使用的一种替代品...

参考莫凡博客进行MNIST数据集的训练，临时记录所使用的代码。import torchimport torch.nn as nnimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as plttorch.manual_seed(1)EPOCH = 1BATCH_SIZE = 50...