optuna.integration.PyTorchLightningPruningCallback optuna.integration PyTorchLightningPruningCallback Methods ~PyTorchLightningPruningCallback.on_validation_end