ImportError: cannot import name 'EPOCH_OUTPUT' from 'pytorch_lightning.utilities.types'

[nanshanvv]

When I run the code I get the following error, how do I fix it?

[J4k3j4k3]

I believe this was trained perhaps on a previous version of lightning? But I'm not sure which one. That epoch_output doesnt exist now

[J4k3j4k3]

Downgrading to pytorch-lightning v1.9,4 fixed that issue for me: pip install pytorch-lightning==1.9.4