# python train_model.py problem.hyp.alpha=0.0 problem/model=dt_net_2d problem=mnist name=mnist_ablation
# python train_model.py problem.hyp.alpha=0.50 problem.hyp.epochs=200 problem.hyp.lr=0.0001 problem.hyp.lr_schedule=[175] problem/model=ff_net_recall_2d problem=mnist name=mnist_ablation
python train_model.py name=cifar10_dt_net_attention_2d problem=mnist problem/model=dt_net_2d_custom problem.model.num_class=10