bottom-up-attention.pytorch
bottom-up-attention.pytorch copied to clipboard
Stuck when extracting features
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.
Have you solved this problem? Can you tell me how to solve it?
I have also encountered this problem. Can you tell me how to solve it