train.ModelCheckpoint

class train.ModelCheckpoint(*args: Any, **kwargs: Any)

Like the PyTorch Lightning ModelCheckpoint callback, but also saves the best inference model with activated post-processing