This repository contains code for validating the theory for the paper "On the Convergence of Shallow Neural Network Training with Randomly Masked Neurons"

run the following command to reproduce data for the corresponding figure:
    Figure 2a: "./run_exp.sh"
    Figure 2b: "./run_heatmap.sh"
    Figure 2c: "./run_heatmap_cat.sh"