Callbacks (radionets.core.callbacks)

Callbacks submodule of radionets.core.

Reference/API

radionets.core.callbacks Module

Classes

CometCallback(name, validation_data, ...)	Callback for logging training metrics and visualizations to Comet ML.
AvgLossCallback()	Callback for tracking and plotting average training and validation losses.
PredictionImageGradient(validation_data, ...)	Callback for computing spatial gradients of model predictions.
GradientCallback(num_epochs, ...)	Callback for gradient and prediction tracking.
SwitchLoss(second_loss, when_switch)	Callback for switching loss functions during training.
SaveTempCallback(model_path)	Callback for saving temporary model checkpoints during training.
Normalize(conf)	Normalization callback for input and target data.
DataAug(*[, after_create, before_fit, ...])	Callback that applies data augmentation using random rotations.
CudaCallback(*[, after_create, before_fit, ...])	Callback to move model to CUDA device before training.

Class Inheritance Diagram