pytorch的神经网络模板。
item可以把tensor数据类型转化为普通的数字。
保存模型的。pth这里是可以自定义的,一般保为pth
小配件:准确率
1代表的是纵向的。
使用gpu进行训练——在数据,模型,损失函数上+cuda即可
第二种使用gpu的方法
pytorch的神经网络模板。
item可以把tensor数据类型转化为普通的数字。
保存模型的。pth这里是可以自定义的,一般保为pth
小配件:准确率
1代表的是纵向的。
使用gpu进行训练——在数据,模型,损失函数上+cuda即可
第二种使用gpu的方法