Duplicated from MLSB/pinder_inference_template
c618657
1
2
3
4
5
6
emb_dim: 32 in_dim: 3 lr: 0.0001 num_heads: 1 num_layers: 4