Pytorch 基本流程
pytorch in action
pytorch复现神经网络(深度学习)模型,一般包含如下三个步骤,本文对各个步骤给出可以参考的代码框架.
数据准备
构建模型
训练与评估
数据准备
train_data (train_data, valid_data)
test_data
构建自己的数据集
继承DataSet
from torch.utils.data import Dataset, D...
原创
2019-10-22 21:35:47 ·
2539 阅读 ·
0 评论