代码复现过程中遇到的问题

运行环境:cuda 9.2 、python 3.6 、torch 0.4.1
Installation部分

  1. 运行./make.sh时报错,权限不够
    解决方法:
cd PPDM-master/src/lib/models/networks/DCNv2
chmod 777 make.sh
./make.sh

2.复制环境后使用pip时出错(pip对应的python路径不对)
解决办法:
which pip
/home/wll/anaconda3/envs/test/bin/pip

which python
/home/wll/anaconda3/envs/test/bin/python

vim /home/wll/anaconda3/envs/test/bin/pip
把#!后面的改成python的路径就好了

3.torch版本换成0.4.1版本
手动下载torch安装包网址:(torch.pypi里面找不到想要的版本)
https://download.pytorch.org/whl/torch_stable.html

Training部分
4.
运行

cd src
python main.py  hoidet --batch_size 112 --master_batch 7 --lr 4.5e-4 --gpus 0,1,2,3,4,5,6,7  --num_workers 16  --load_model ../models/ctdet_coco_dla_2x.pth --image_dir images/train2015 --dataset hico --exp_id hoidet_hico_dla

出现如下错误(加载模型时出错)
在这里插入图片描述解决办法:将/home/wanglulu/PPDM_master/src/lib/models/networks/pose_dia_dcn.py 中的pretained的True改为False

5.如何解决RuntimeError: CUDA error: out of memory?(爆显存问题)
将batch size 改小,一般设置为2的幂次,32,64,128,…

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值