athena.nll.NonlinearLevelSet.load_forward

NonlinearLevelSet.load_forward(infile, n_params)

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.