Skip to main content

classy.utils.lightning

Functions​

load_classy_module_from_checkpoint​

def load_classy_module_from_checkpoint(
    checkpoint_path: str,
) ‑> ClassyPLModule
##

Load a PL module from a checkpoint path only. Infer the model to load from the dumped hydra conf

Args​

checkpoint_path (str):

Returns​

pl.LightningModule

load_prediction_dataset_conf_from_checkpoint​

def load_prediction_dataset_conf_from_checkpoint(
    checkpoint_path: str,
) ‑> omegaconf.dictconfig.DictConfig
##

Load a dataset conf from a checkpoint path only, inferring it from the dumped hydra conf

Args​

checkpoint_path (str):

Returns​

Dict

load_training_conf_from_checkpoint​

def load_training_conf_from_checkpoint(
    checkpoint_path: str,
    post_trainer_init: bool = False,
) ‑> omegaconf.dictconfig.DictConfig
##