opto.trace.modules

Contents

`opto.trace.modules`#

Module Contents#

Classes#

Module is a ParameterContainer which has a forward method.

Functions#

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

API#

model(cls)[source]#: Wrap a class with this decorator. This helps collect parameters for the optimizer. This decorated class cannot be pickled.

class Module[source]#

Bases: opto.trace.containers.ParameterContainer

Module is a ParameterContainer which has a forward method.

abstract forward(*args, **kwargs)[source]#

save(file_name)[source]#: Save the parameters of the model to a file.

load(file_name)[source]#: Load the parameters of the model from a file.