python figures.py \
--experiment_name track-test \
--run_name track-all \
--output train_vs_val_loss \