pytorch学习笔记

pytorch的神经网络模板。

pytorch学习笔记_第1张图片
pytorch学习笔记_第2张图片
pytorch学习笔记_第3张图片

pytorch学习笔记_第4张图片

item可以把tensor数据类型转化为普通的数字。

pytorch学习笔记_第5张图片
pytorch学习笔记_第6张图片
pytorch学习笔记_第7张图片
pytorch学习笔记_第8张图片

保存模型的。pth这里是可以自定义的,一般保为pth

小配件:准确率

pytorch学习笔记_第9张图片

1代表的是纵向的。

pytorch学习笔记_第10张图片

使用gpu进行训练——在数据,模型,损失函数上+cuda即可

pytorch学习笔记_第11张图片

pytorch学习笔记_第12张图片
pytorch学习笔记_第13张图片

第二种使用gpu的方法

pytorch学习笔记_第14张图片

你可能感兴趣的:(pytorch,学习,深度学习)