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.