2024年8月7日
【深度学习】用Pytorch完成MNIST手写数字数据集的训练和测试
摘要： 模型训练相关 思路： 导入数据集（对数据集转换为张量）加载数据集（使数据集成为可以进行迭代）搭建卷积模型进行模型训练（每训练一轮查看一次在测试集上的准确率）使用tensorboard进行可视化保存训练后的模型加载训练好的模型进行测试. 选择的模型结构 训练模型和评估模型 在conda命令行输入 te 阅读全文
posted @ 2024-08-07 17:48 Lnnau 阅读(626) 评论(0) 推荐(0)