python3 main.py dataset=helmholtz exp.lr=0.01 model.input_bc=1 model.input_gradtheta=1 The available config are summarized in the params.txt file. To replicate Figure 2, first train your model: ...