Closed abhilashi closed 4 years ago
It works well when I specify a single GPU.
python tools/extract_features.py --test_data=dupes_data --model_name=r2plus1d --model_depth=34 --clip_length_rgb=32 --gpus=0 --batch_size=4 --load_model_path=./trained/r2.5d_d34_l32_ft_sports1m.pkl --output_path=my_features.pkl --features=softmax,final_avg,video_id --sanity_check=0 --get_video_id=1 --use_local_file=1 --num_labels=400
...
[swscaler @ 0x7e90e4162300] Warning: data is not aligned! This can lead to a speedloss
0/1 iterations
INFO:feature_extractor:0/1 iterations
Read 'softmax' with shape (4, 400)
INFO:feature_extractor:Read 'softmax' with shape (4, 400)
Read 'final_avg' with shape (4, 512, 1, 1, 1)
INFO:feature_extractor:Read 'final_avg' with shape (4, 512, 1, 1, 1)
Read 'video_id' with shape (4,)
INFO:feature_extractor:Read 'video_id' with shape (4,)
Writing to my_features.pkl
seems like memory issue.
Anyone else who has seen this issue? CC @dutran