File size: 443 Bytes
2ce7b1a |
1 2 3 4 5 |
python run_train.py --C_init=complex_normal --batchnorm=True --bidirectional=True \
--blocks=16 --bn_momentum=0.9 --bsz=32 --d_model=128 --dataset=pathx-classification \
--dt_min=0.0001 --epochs=75 --jax_seed=6429262 --lr_factor=3 --n_layers=6 \
--opt_config=BandCdecay --p_dropout=0.0 --ssm_lr_base=0.0006 --ssm_size_base=256 \
--warmup_end=1 --weight_decay=0.06 |