ModuleWrapperBase¶
- class flash.core.model.ModuleWrapperBase[source]¶
The
ModuleWrapperBase
is a base for classes which wrap aLightningModule
or an instance ofModuleWrapperBase
.This class ensures that trainer attributes are forwarded to any wrapped or nested
LightningModule
instances so that nested calls to.log
are handled correctly. TheModuleWrapperBase
is also stateful, meaning that aDataPipelineState
can be attached. Attached state will be forwarded to any nestedModuleWrapperBase
instances.