- 博客(5)
- 收藏
- 关注
RSS订阅原创 pytorch同时在多个GPU上训练模型nn.Dataparallel
使用方法：model为自己的模型 device_ids = [0, 1] model = torch.nn.DataParallel(model, device_ids=device_ids) 这样就可以在0号和1号GPU上跑了
2022-03-17 16:57:12
1394
原创 将a*b*c三维矩阵进行0填充变成m*n*c三维矩阵的方法
矩阵A为a*b*c 设置填充矩阵 pad1 = np.zeros((m,n-b,c)) A = np.column_stack(A,pad1) 此时A维度为a*m*c pad2 = np.zeros((m-a,n,c)) A = np.row_stack(A,pad2) A矩阵为m*n*c
2022-02-28 17:27:01
368
原创 linux服务器用tensorboard打开event.out.tfevents文件被拒绝访问的解决办法
1.event.out.tfevents文件保存到本地，再用tensorboard打开 2. ssh -L 16006:127.0.0.1:6006 account@server.address 在连接时将本地端口号与服务器相对应，在本地时访问http://127.0.0.1:16006/即可跳转到tensorboard页面。
2022-01-05 16:43:17
2364
2
原创 python数据太长变成省略号，保存也是省略号的解决方案
存储中间数据突然发现数据变成了Ellipis，存储的一批数据中有正常的，有省略号的，经观察发现有省略号的数据是因为数据太长了，导致输出为省略号，在数据前加上这句话就可以使数据全部保存了。 np.set_printoptions(threshold=sys.maxsize) ...
2022-01-05 16:40:43
3342
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人