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