python main_avel.py \
--seed 87 \
--mode train \
--model PreFM_Net \
--model_name model_PreFM_On_AVEL_reproduce \
--gpu 0 \
--batch_size 128 \
--val_batch_size 60 \
--epochs 60 \
--pd_dir_train ./data/train_pd.csv \
--pd_dir_val ./data/val_pd.csv \
--pd_dir_test ./data/test_pd.csv \
--audio_dir /your_audio_input_dir/ \
--visual_dir /your_video_input_dir/ \
--label_dir /your_label_input_dir/ \
--label_train /your_label_input_dir/train \
--label_val /your_label_input_dir/val \
--label_test /your_label_input_dir/test \
--f_label_train /your_feature_label_input_dir/train \
--f_label_val /your_feature_label_input_dir/val \
--f_label_test /your_feature_label_input_dir/test \
--optimizer adamw \
--weight_decay 1e-4 \
--scheduler warm_up_cos_anneal \
--warm_up_epoch 10 \
--grad_norm 1.0 \
--lr 1e-3 \
--lr_min 1e-5 \
--beta1 0.5 \
--eps 1e-8 \
--hidden_dim 256 \
--nhead 8 \
--ff_dim 1024 \
--num_layers 4 \
--norm_where post_norm \
--future_length 5 \


# bash scripts/train_avel.sh | tee ./logs/prefm_on_avel_train.txt