hstu.yaml 189 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
model: HSTU
n_layers: 2
n_heads: 4
item_embedding_size: 512
hstu_embedding_size: 512
hidden_dropout_prob: 0.1
attn_dropout_prob: 0.1
hidden_act: 'silu'
enable_relative_attention_bias: True