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.