File size: 494 Bytes
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 32 \
--dataset steam_data \
--data_dir data/ref/steam \
--cans_num 20 \
--prompt_path ./prompt/game.txt \
--rec_embed SASRec \
--llm_tuning moelora \
--llm_path $LLM_PATH \
--rec_model_path ./rec_model/steam.pt \
--ckpt_path ./checkpoints/steam.ckpt \
--output_dir ./output/steam/ \
--log_dir steam_logs \
--lr_warmup_start_lr 5e-6 \
--lr 5e-4 \
--lr_decay_min_lr 5e-6 \
--max_epochs 5
|