The reduce function in-built to the Result class only gets called if the epoch_end methods aren’t overridden, and that check (whether the epoch_end method is overridden) happens in evaluation_loop.py, so I don’t see a good way of getting around this while also using PL’s inbuilt aggregation without just overriding the epoch_end method and aggregating it yourself.
2 Likes