opto.trace.modules.Module

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.

parameters()

Return a flattned list of all the parameters in the model's parameters_dict, useful for optimization.

parameters_dict()

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)#