athena.nll.NonlinearLevelSet.load_forward

NonlinearLevelSet.load_forward(infile, n_params)[source]

Load the forward map for inference.

Parameters
  • infile (str) – filename of the saved net to load. See notes below.

  • n_params (int) – number of input parameters.

Note

A common PyTorch convention is to save models using either a .pt or .pth file extension.