temporal-shift-module
temporal-shift-module copied to clipboard
How to fine-tune with 'TSM_kinetics_RGB_mobilenetv2_shift8_blockres_avg_segment8_e100_dense.pth'
I want to fine-tune on my dataset with 'TSM_kinetics_RGB_mobilenetv2_shift8_blockres_avg_segment8_e100_dense.pth'
I can train successfully with 'TSM_kinetics_RGB_mobilenetv2_shift8_blockres_avg_segment8_e100_dense.pth' , but when I train with command:
python main.py mydata RGB
--arch mobilenetv2 --num_segments 8
--gd 20 --lr 0.001 --lr_steps 10 20 --epochs 25
--batch-size 8 -j 16 --dropout 0.8 --consensus_type=avg --eval-freq=1
--shift --shift_div=8 --shift_place=blockres
--tune_from=pretrained/TSM_kinetics_RGB_mobilenetv2_shift8_blockres_avg_segment8_e100_dense.pth
I got error:
Traceback (most recent call last):
File "main.py", line 378, in
What should I change?
Are you training on GPU? I remember having a DataParallel issue when I didn't initialize my GPUs.
Any solution for this problem?
how to download TSM_kinetics_RGB_mobilenetv2_shift8_blockres_avg_segment8_e100_dense.pth??