python train_dynamics.py --dataset life -s --epochs 1000 --bn --lr 0.01
python train_dynamics.py --dataset waves --epochs 1000 --bn --lr 0.01
python train_dynamics.py --dataset spirals --epochs 1000 --bn --lr 0.01