Table of Contents

Shortcuts

pytorch_lightning.trainer.model_hooks module¶

class pytorch_lightning.trainer.model_hooks.TrainerModelHooksMixin[source]¶

Bases: abc.ABC

abstract get_model()[source]¶

Warning: this is just empty shell for code implemented in other class.

Return type: LightningModule

has_arg(f_name, arg_name)[source]¶

is_function_implemented(f_name, model=None)[source]¶

is_overridden(method_name, model=None)[source]¶

Return type: bool