| 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 | |