Shortcuts

ModuleWrapperBase

class flash.core.model.ModuleWrapperBase[source]

The ModuleWrapperBase is a base for classes which wrap a LightningModule or an instance of ModuleWrapperBase.

This class ensures that trainer attributes are forwarded to any wrapped or nested LightningModule instances so that nested calls to .log are handled correctly. The ModuleWrapperBase is also stateful, meaning that a DataPipelineState can be attached. Attached state will be forwarded to any nested ModuleWrapperBase instances.

Read the Docs v: stable
Versions
latest
stable
0.8.1
0.8.0
0.7.5
0.7.4
0.7.3
0.7.2
0.7.1
0.7.0
0.6.0
0.5.2
0.5.1
0.5.0
0.4.0
0.3.2
0.3.1
0.3.0
0.2.3
0.2.2
0.2.1
0.2.0
0.1.0post1
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.