vbals的博客

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

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