13 lines
599 B
Bash
13 lines
599 B
Bash
|
python train_pytorchvideo_x3d.py \
|
||
|
|
--csv_dir /home/ubuntu/data/fish/fish_action_videos \
|
||
|
|
--path_prefix /home/ubuntu/data/fish/fish_action_videos \
|
||
|
|
--model x3d_m \
|
||
|
|
--pretrained \
|
||
|
|
--batch_size 20 \
|
||
|
|
--epochs 100 \
|
||
|
|
--train_clips_per_video 10 \
|
||
|
|
--num_workers 4 \
|
||
|
|
--lr 3e-4 \
|
||
|
|
--scheduler cosine --warmup_epochs 2 --min_lr 1e-5 \
|
||
|
|
--output_dir /home/ubuntu/projects/FishAction/checkpoints/ptv_x3d_m_class3_epoch100_split1min_newscared --log_interval 2 --val_sampling constant_clips_per_video --val_clips_per_video 2
|