File size: 519 Bytes
9f13819 8ed221f 9f13819 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
python main.py \
--mode test \
--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_path ./checkpoints/movielens.ckpt \
--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
|