yangpei1993的专栏

原创 神经网络-pytorch常用包介绍（一）

神经网络-pytorch常用包介绍（一）pytorch学习笔记：从数据处理、加载、模型建立、定义损失函数、定义反向传播方法、训练、保存模型、加载模型。 首先看一下常用的包：from __future__ import print_functionimport argparse import osimport torchfrom torch.utils.data import D...