llama_id.yaml 222 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
10
11
12
13
model: LLMIDRec
user_pretrain_dir: user_pretrain_dir
user_llm_init: False
use_ft_flash_attn: True
item_embed_dim: 512

show_progress: True

# better to use cosine scheduler
scheduler_args: {
  type: cosine,
  warmup: 0.0
}