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.