随笔分类 -  pytorch学习

摘要:代码地址:https://github.com/jwyang/faster-rcnn.pytorch 1.fasterRCNN.train():这个不是让网络进行训练,而是让module in training mode,有些module在traing model和testing model下不同, 阅读全文
posted @ 2018-12-07 18:46 有梦就要去实现他 阅读(2900) 评论(0) 推荐(0) 编辑
摘要:1.tensor的view函数: view(*args) → Tensor 返回一个有相同数据但大小不同的tensor。 返回的tensor必须有与原tensor相同的数据和相同数目的元素,但可以有不同的大小。一个tensor必须是连续的contiguous()才能被查看。 其实就是类似caffe的 阅读全文
posted @ 2018-12-07 15:56 有梦就要去实现他 阅读(333) 评论(0) 推荐(0) 编辑
摘要:代码里面用tensorboard保存了训练的日志在logs目录里面 用tensorboard命令打开日志目录:tensorboard --logdir="./logs/" 会显示一个网址: TensorBoard 1.6.0 at http://bnrcDL:6006 (Press CTRL+C t 阅读全文
posted @ 2018-12-06 15:03 有梦就要去实现他 阅读(1308) 评论(0) 推荐(0) 编辑
摘要:使用PyTorch设置多线程(threads)进行数据读取(DataLoader),其实是假的多线程,他是开了N个子进程(PID都连着)进行模拟多线程工作,所以你的程序跑完或者中途kill掉主进程的话,子进程的GPU显存并不会被释放,需要手动一个一个kill才行,具体方法描述如下: 1.先关闭ssh 阅读全文
posted @ 2018-12-05 12:23 有梦就要去实现他 阅读(4005) 评论(0) 推荐(0) 编辑
摘要:只有专业版的才能同步服务器 按照这个来:https://zhuanlan.zhihu.com/p/35067462 3、然后配置映射信息 local path是自己的工程的本地目录路径, Deploy path on server 这里填写相对于root path的目录. 这个地方注意:Deploy 阅读全文
posted @ 2018-12-03 14:13 有梦就要去实现他 阅读(169) 评论(0) 推荐(0) 编辑
摘要:服务器是的cuda9,python3.6,unbuntu16 pytorch之前的版本:http://pytorch.org/get-started/previous-versions/ 之前下载的cu90/torch-0.4.0-cp36-cp36m-linux_x86_64.whl,用pip3 阅读全文
posted @ 2018-12-02 22:03 有梦就要去实现他 阅读(2928) 评论(0) 推荐(0) 编辑
摘要:pytorch版本的faster和fpn https://github.com/jwyang/faster-rcnn.pytorch https://github.com/jwyang/fpn.pytorch 阅读全文
posted @ 2018-12-02 20:56 有梦就要去实现他 阅读(496) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示