Hi @xen0f0n
So, output transform argument of Loss metric : https://pytorch.org/ignite/metrics.html#ignite.metrics.Loss 6
does not work in your case ?
If it’s the case, please, provide a minimal code snippet to understand the issue. Thanks!