JacobJiang的博客

import torchfrom torch import nnimport torchvision.datasets as dsetsimport torchvision.transforms as transformsimport matplotlib.pyplot as pltimport os# Hyper ParametersEPOCH = 10BATCH_SIZE ...

import torchimport torch.utils.data as Dataimport torchvisionimport matplotlib.pyplot as pltimport os# Hyper ParametersEPOCH = 1BATCH_SIZE = 50LR = 0.001DOWNLOAD_MNIST = False# 拥有两层卷基层+输出...

 使用torch完成一个简单的回归 1、建立一个三层神经网络：输入层，隐藏层，输出层。这里建立网络的方法可以参考：torch建立网络的两种方式 2、生成一些随机的点（一个二次函数+噪声），用于回归：如下所示 3、训练效果 4、代码如下import torchimport matplotlib.pyplot as plt...

1、torch 与 numpy 之间的转化import torchimport numpy as npnp_data = np.arange(6).reshape((2, 3)) # numpy arraytorch_data = torch.from_numpy(np_data) # torch tensortorch_data2np = torch_data.numpy() ...

1、PyTorch 安装PyTorch 官网已经有很详细的安装方法：https://pytorch.org/Linux and Windows：1、查看自己电脑安装的cuda版本Windows：开始菜单->运行->输入cmd（或者Windows+R,然后输入cmd），输入nvcc -VLinux：打开终端（ctrl+alt+t）输入相同的指令即可。2、使用 pip 安装...