To plot the figures in Figure 4
please run_all.sh to save the models output in checkpoints
Then inside plot_main.py choose the learning rate, batch size, and c to plot the experiments.
