athena.nll.NonlinearLevelSet.save_backward¶

NonlinearLevelSet.save_backward(outfile)[source]¶

Save the backward 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.