摘要: 一、训练流程 1. 准备数据集 2. 计算训练集和测试集的长度 2.1 输出训练集和测试集的长度 3. 利用DataLoader来加载数据集 4. 搭建神经网络 在另外的页面搭建好神经网络模型(test27_2_model) 在当前页面进行调用(test27_1_train) 5.创建网络模型 6. 阅读全文
posted @ 2022-05-25 21:25 Trouvaille_fighting 阅读(219) 评论(0) 推荐(0)
摘要: 一、案例 1.保存方式1 保存模型结构+模型参数 2.保存方式1的陷阱 自己创建的模型,在加载的时候,需要import 3.保存方式2 保存模型参数(官方推荐) 比较1和2两种方式保存的内容的大小 terminal执行 ls -all 4.完整代码 模型保存 import torch import 阅读全文
posted @ 2022-05-25 11:18 Trouvaille_fighting 阅读(103) 评论(0) 推荐(0)