python main_avvp.py \
--seed 87 \
--mode train \
--model PreFM_Net \
--model_name model_PreFM_On_AVVP_reproduce \
--gpu 0 \
--batch_size 128 \
--val_batch_size 60 \
--epochs 60 \
--pd_dir_train ./data_online/train_pd_online.csv \
--pd_dir_val ./data_online/val_pd_online.csv \
--pd_dir_test ./data_online/test_pd_online.csv \
--audio_dir /your_audio_input_dir/ \
--visual_dir /your_visual_input_dir/ \
--st_dir /your_st_input_dir/st \
--label_dir /your_label_input_dir/ \
--f_label_dir /your_feature_label_input_dir/ \
--optimizer adamw \
--weight_decay 1e-4 \
--scheduler warm_up_cos_anneal \
--warm_up_epoch 10 \
--grad_norm 1.0 \
--lr 5e-4 \
--lr_min 5e-6 \
--beta1 0.5 \
--eps 1e-8 \
--hidden_dim 128 \
--nhead 8 \
--ff_dim 512 \
--num_layers 4 \
--norm_where post_norm \
--current_length 10 \
--future_length 5 \
--interval 6 \

# bash scripts/train_on.sh | tee ./logs/prefm_on_avvp_train.txt