- 博客(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
1269
原创 将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*cpad2 = np.zeros((m-a,n,c))A = np.row_stack(A,pad2)A矩阵为m*n*c
2022-02-28 17:27:01
256
原创 linux服务器用tensorboard打开event.out.tfevents文件被拒绝访问的解决办法
1.event.out.tfevents文件保存到本地，再用tensorboard打开2.ssh -L 16006:127.0.0.1:6006 [email protected]在连接时将本地端口号与服务器相对应，在本地时访问http://127.0.0.1:16006/即可跳转到tensorboard页面。
2022-01-05 16:43:17
2014
2
原创 python数据太长变成省略号，保存也是省略号的解决方案
存储中间数据突然发现数据变成了Ellipis，存储的一批数据中有正常的，有省略号的，经观察发现有省略号的数据是因为数据太长了，导致输出为省略号，在数据前加上这句话就可以使数据全部保存了。np.set_printoptions(threshold=sys.maxsize)...
2022-01-05 16:40:43
2977
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人