niteshade.utils.save_model

niteshade.utils.save_model(model, filename)

Save an object as a binary .pickle file.

Parameters

• model (nn.Module) – model to save

• filename (str) – name of file to save model in (excluding extension)