Skip to main content

classy.pl_callbacks.prediction

Classes​

EvaluationPredictionCallback​

class EvaluationPredictionCallback()

##

FileDumperPredictionCallback​

class FileDumperPredictionCallback()

##

PredictionCallback​

class PredictionCallback()

##
Subclasses (3)

PredictionPLCallback​

class PredictionPLCallback(pytorch_lightning.callbacks.base.Callback)

##

Abstract base class used to build new callbacks.

Subclass this class and override any of the relevant hooks

__init__​

def __init__(
    path: str,
    prediction_dataset_conf: omegaconf.dictconfig.DictConfig,
    on_result: Dict[str, omegaconf.dictconfig.DictConfig],
    settings: List[Dict[str, Any]],
)
##

on_validation_epoch_start​

def on_validation_epoch_start(
    self,
    trainer: pytorch_lightning.trainer.trainer.Trainer,
    model: ClassyPLModule,
) ‑> None
##

Called when the val epoch begins.

WANDBLoggerPredictionCallback​

class WANDBLoggerPredictionCallback()

##