bottom-up-attention.pytorch icon indicating copy to clipboard operation
bottom-up-attention.pytorch copied to clipboard

Stuck when extracting features

Open dongfengxijian opened this issue 1 year ago • 2 comments

When I use the extract_features.py to extract features, the process seem be stuck. I tired the two commands:

python3 extract_features.py --mode caffe
--num-cpus 2 --gpus '0,1,2,3'
--extract-mode roi_feats
--min-max-boxes '20, 20'
--config-file configs/caffe/test-caffe-r101.yaml
--image-dir './datasets/demo/' --bbox-dir './output/features/' --out-dir './output/bbox/' --fastmode

python3 extract_features.py --mode caffe
--num-cpus 0 --gpus '0,1,2,3'
--extract-mode roi_feats
--min-max-boxes '10, 100'
--config-file configs/caffe/test-caffe-r101.yaml
--image-dir './datasets/demo/' --bbox-dir './output/features/' --out-dir './output/bbox/'

However, when I removed the --fastmode, it was still stuck. Addition, I can find the process through top command but I can't find it through nvidia-smi command. 微信截图_20230526224537

dongfengxijian avatar May 26 '23 14:05 dongfengxijian

Have you solved this problem? Can you tell me how to solve it?

ther-0319 avatar Nov 20 '23 08:11 ther-0319

I have also encountered this problem. Can you tell me how to solve it

1MurasaKi avatar Dec 12 '23 10:12 1MurasaKi