File size: 439 Bytes
2ce7b1a |
1 2 3 4 5 |
python run_train.py --C_init=trunc_standard_normal --batchnorm=True --bidirectional=True \
--blocks=8 --bn_momentum=0.9 --bsz=64 --d_model=192 \
--dataset=pathfinder-classification --epochs=200 --jax_seed=8180844 --lr_factor=5 \
--n_layers=6 --opt_config=standard --p_dropout=0.05 --ssm_lr_base=0.0009 \
--ssm_size_base=256 --warmup_end=1 --weight_decay=0.03 |