Mnist手写数据集的模型搭建及GPU训练(基于pytorch)
1. 用意pytorch看了有一段时间了,可是怎么熟练操作pytorch呢,于是就用手写数据集来办事,键盘滚起来!四颗GPU训练两个类完成训练代码直接copy过去,立即跑通,但电脑上必须有GPU代码直接copy过去,立即跑通但电脑上必须有GPU代码直接copy过去,立即跑通但电脑上必须有GPU2.结果展示2.1 四颗GPU2.2 训练过程2.3 训练结果3.代码如下import osimport randomimport torchimport numpy as np
原创
2020-11-19 22:01:29 ·
838 阅读 ·
0 评论