pytorch模型加载与保存

2020-03-25  本文已影响0人  空谷幽兰_c411

pytorch模型训练流程

  1. 配置超参数： epoch，learning_rate
  2. 构建数据集：训练集，验证集，测试集。
    并构建相应的data_loader
    3.搭建模型：继承nn.Module，编写init方法和forword方法
  3. 配置损失函数，优化器
  4. 搭建训练过程和测试过程例程
  5. 运行main训练
上一篇 下一篇

猜你喜欢

热点阅读