Paper figures
This folder holds code for creating Fig. 1B and all of the plots in Fig. 2B. These figures are programmatically generated, but not part of a particular training/benchmarking pipeline. For all other figures in the paper, any code used to generate figures will be in the appropriate training/, data/, or benchmarking/ subfolder.
Figure 1B
To recreate the plotted circles in Figure 1B, go to the fig1 folder and enter python data_circles.py.
Figure 2B
To recreate the masking rate and scheduler plots in figure 2B, go to the fig2 folder and enter python mask_rate_plots.py.