code
guoqiangszu
这个作者很懒,什么都没留下…
展开
-
Pytorch网络结构可视化方法汇总(三种实现方法详解)
前言:在深度学习的时候,能够很好的绘制出整个网络的模型图,这是非常方便的,当前流行的tensorflow和pytorch都有很多种方法可以绘制模型结构图,特在此总结如下:tensorflow的模型结构可视化方法:(1)使用自带的tensorboard(不直观)(2)使用netron工具打开(.pd 或者是.meta文件)(3)第三方库CNNGraph(https://github.com/huachao1001/CNNGraph)(4)tensorspace.js (这个比较高级,没..转载 2020-05-11 21:05:48 · 4164 阅读 · 3 评论 -
Pytorch使用torch.nn.DataParallel()在Linux服务器上多GPU训练
前期我自己也在网上也找了很多相关的文章,但是都不能用。后来结合Pytorch中多GPU训练这篇文章,加上自己的尝试,终于成功在服务器上使用多gpu训练成功。首先,使用代码os.environ["CUDA_VISIBLE_DEVICES"] = "2, 3, 4" 这句代码能够告诉程序接下来要使用的gpu编号是哪几块。比如服务器上有8块gpu,编号为0,1, 2,……,7,你只想用其中的2,...原创 2020-05-01 16:35:39 · 1405 阅读 · 4 评论 -
PyTorch中指定GPU块号的方法
1、 改变系统变量使得仅目标显卡可见export CUDA_VISIBLE_DEVICES=0,1,2...os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2...'CUDA_VISIBLE_DEVICES=0,1,2... python xxx.py ,设置该程序可见的gpu这里的“0,1,2”指的是gpu的编号,一般情况下是从0开始。...原创 2020-03-18 17:21:58 · 3401 阅读 · 0 评论 -
python对文件进行批量重命名
以mat文件为例原文件名如下图:现在变成下图:代码如下所示:import osif __name__ == '__main__': root = "E:/data" path= os.path.join(root, 'UCF-QNRF_pro/train_data/ground_truth') count = 0 file_list = o...原创 2019-10-13 20:16:22 · 161 阅读 · 0 评论