torch
猫老壳
这个作者很懒,什么都没留下…
展开
-
torch多GPU模型的训练与保存
若模型采用多GPU训练,则在模型保存时:torch.save(model.module.state_dict(), model_out_path)若单GPU则:torch.save(mode.state_dict(), model_out_path)原创 2019-08-14 16:34:19 · 824 阅读 · 0 评论 -
使用CSV文件作为标签格式的输入处理
CSV文件的读取。import pandas as pddf = pd.read_csv("train.csv")print(df.head()) # 打印head和前几个数据# 遍历获取值,则:for data in df.values: print(data) # 列表import pandas as pddf = pd.read_csv("train.csv")d...原创 2019-08-25 17:06:21 · 1822 阅读 · 0 评论 -
torch中的几种乘法。torch.mm, torch.mul, torch.matmul
一、点乘点乘都是broadcast的,可以用torch.mul(a, b)实现,也可以直接用*实现。>>> a = torch.ones(3,4)>>> atensor([[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]])>>> b = torc...原创 2019-09-10 10:44:12 · 25312 阅读 · 3 评论 -
(采坑之)torch在网络中自定义一个变量进行多GPU训练
在torch中的卷积和全连接层等都可以调用nn.Conv2d、nn.Linear等函数解决,但最近需要在网络中实现点乘,于是自定义了一个权重并进行训练。之前一直在单块GPU上没遇到问题,今天在多块GPU上训练报错如下:File "**.py", line 72, in forward a = torch.mul(a_sum, self.weights2)RuntimeError: bi...原创 2019-09-22 08:59:33 · 1405 阅读 · 0 评论