Files
FishServer/FishAction/train.sh
2026-04-08 19:32:23 +08:00

13 lines
599 B
Bash
Executable File

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