test_rgb.sh cannot work. And get an error:RuntimeError: shape '[0, 8, 64, 28, 28]' is invalid for input of size 50176

[SceneRec]

Thanks for your beautiful work.
I trained the .sh file python main.py something-v1 RGB --arch BNInception \ --num_segments 8 --consensus_type avg \ --batch-size 16 --iter_size 2 --dropout 0.5 \ --lr 0.01 --warmup 10 --epochs 60 --eval-freq 5 \ --gd 20 --run_iter 1 -j 16 --npb --gsm I got a model "something-v1_BNInception_avg_segment8_checkpoint.pth.tar" the test_rgb.sh: python test_models.py somethong-v1 RGB models/something-v1_BNInception_avg_segment8_checkpoint.pth.tar \ --arch BNInception --crop_fusion_type avg --test_segments 8 --test_crops 1 --num_clips 1 --gsm

When I run the test._rgb.sh, the error arised ** File "/data/users/xuyang/xuyang/Downloads/GSM-master/gsm.py", line 31, in forward x = x.view(batchSize, self.num_segments, shape).permute(0, 2, 1, 3, 4).contiguous() RuntimeError: shape '[0, 8, 64, 28, 28]' is invalid for input of size 50176*

I try some ways, but the error still asised. Please guide me. Look forward to your reply!

[swathikirans]

Try running the evaluation code on a single GPU.