DefeatedCrab47 since we can't fix this issue ourselves, we'd appreciate your feedback on https://github.com/PyTorchLightning/pytorch-lightning-bolts/pull/86 PR in the PyTorchLightning/pytorch-lightning-bolts, just to let the guys there know someone's waiting for it :)