David Pitt
David Pitt
Improves docstring for `Callback` in `neuralop.training.callbacks.py` and adds a small example to showcase the simplest case of a Callback in training. Also fixes formatting for the IFNO example to be...
Adds a minimal example training the GINO on our `CarCFDDataset`.
In line with adding more PINN tricks, this is a quick PR that adds shake-shake loss regularization to `neuralop.losses.meta_losses` as `ShakeShakeLoss`.
Refactors some old repeated logic out of the training recipe scripts and puts it in `neuralop/training/torch_setup.py`
I'm getting an error on Ubuntu when installing torch_scatter via `conda install pytorch-scatter -c pyg`. No error at install time, but when I import torch_scatter, it seems like it's compiled...