How to apply multiple GPUs on not `training_step`?

[justusschock]

That’s great new, thanks for sharing with everybody!