train_movielens.sh

python main.py \
--mode train \
--router share \
--gating Dense \
--batch_size 4 \
--accumulate_grad_batches 16 \
--dataset movielens_data \
--data_dir data/ref/movielens \
--cans_num 20 \
--prompt_path ./prompt/movie.txt \
--rec_embed SASRec \
--llm_tuning moelora \
--llm_path $LLM_PATH \
--rec_model_path ./rec_model/movielens.pt \
--ckpt_dir ./checkpoints/movielens/ \
--output_dir ./output/movielens/ \
--log_dir movielens_logs \
--lr_warmup_start_lr 8e-6 \
--lr 8e-4 \
--lr_decay_min_lr 8e-6 \
--max_epochs 5