We train the model using an Adam optimizer with the default hyperparameters and a learning rate of 0.001 for 500, 000 epochs. The loss plots are presented in Figure .12. The model is trained with N train = 52 samples, with a batch size of 12, and tested on N test = 12. Three representative samples are presented in Figure to demonstrate the performance of the DeepONet model.