athena.nll.NonlinearLevelSet.save_forward¶

NonlinearLevelSet.save_forward(outfile)[source]¶

Save the forward map for future inference.

Parameters: outfile (str) – filename of the net to save. Use either .pt or .pth. See notes below.

Note

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