yjxiong / tsn-pytorch

Temporal Segment Networks (TSN) in PyTorch
BSD 2-Clause "Simplified" License
1.07k stars 308 forks source link

weried phenomenon about using main.py train ucf101 dataset #122

Closed dandingol03 closed 3 years ago

dandingol03 commented 3 years ago

Hi, @yjxiong i use the main.py script to train ucf101 , but it seems like the script is just loading the rawframe, and never begin to train the model. Here is my setting:

python main.py ucf101 RGB \ --arch BNInception --num_segments 3 \ --gd 20 --lr 0.001 --lr_steps 30 60 --epochs 80 \ -b 120 -j 8 --dropout 0.8 \ --snapshot_pref ucf101bninception

Thanks in advance~

dandingol03 commented 3 years ago

i figure out why, because i use the google colab, it needs a long time to load data. After that, it begin to train