opto.trace.modules.Module#
- class Module[source]#
Module is a ParameterContainer which has a forward method.
Methods
forward
(*args, **kwargs)load
(file_name)Load the parameters of the model from a file.
Return a flattned list of all the parameters in the model's parameters_dict, useful for optimization.
Return a dictionary of all the parameters in the model, including both trainable and non-trainable parameters.
save
(file_name)Save the parameters of the model to a file.
- __init__()#
- __new__(**kwargs)#