opto.trace.modules

Module

Bases: ParameterContainer

Module is a ParameterContainer which has a forward method.

forward

forward(*args, **kwargs)

save

save(file_name: str)

Save the parameters of the model to a pickle file.

load

load(file_name)

Load the parameters of the model from a pickle file.

model

model(cls)

Wrap a class with this decorator. This helps collect parameters for the optimizer. This decorated class cannot be pickled.