niteshade.utils.load_model

niteshade.utils.load_model(filename)

Load a binary file containing a neural network.

Parameters

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

Returns

Trained model inheriting nn.Module with learnable parameters

Return type

model (nn.Module)