Train.predict() call in callback raises an error

[whaowhao]

I have having the same error as here where calling trainer.predict inside a custom callback raises error

[tchaton]

Dear @whaowhao.

Using trainer.preditct on_epoch_end within a Callback is quite hacky and it fails as it is not used properly.

I would advise you to use Loop Customization instead. Here is an example with KFold: https://github.com/PyTorchLightning/pytorch-lightning/blob/master/pl_examples/loop_examples/kfold.py#L158

Best,
T.C