export_model_code: "name (str)" arg listed in docstring, but not usable
current docstring for torch_liberator.export_model_code:
torch_liberator.export_model_code(
dpath,
model,
initkw=None,
export_modules=[],
)
Docstring:
Exports the class used to define a pytorch model as a new python module.
Exports the minimum amount of code needed to make a self-contained Python
module defining the pytorch model class. This exports the actual source
code. The advantage of using this over pickle is that the original code can
change arbitrarilly because all dependencies on the original code are
removed in the exported code.
Args:
dpath (str): directory to dump the model
model (tuple or type or object): class or class instance (e.g. torch.nn.Module)
** name (str): name to use for the file (defaults to the classname)**
initkw (dict): if specified, creates the function `make`, which
initializes the network with the specific arguments.
export_modules (List[str]): A list of modules that the exported code
should not depend on. Any code referenced from these modules will
be statically extracted and copied into the model definition.
Note that this feature is experimental.