schnetpack.train

Callbacks

ModelCheckpoint	Like the PyTorch Lightning ModelCheckpoint callback, but also saves the best inference model with activated post-processing
PredictionWriter	Callback to store prediction results using torch.save.

Scheduler

ReduceLROnPlateau

Extends PyTorch ReduceLROnPlateau by exponential smoothing of the monitored metric