Investigate metrics configuration in Lightning

[tizianocitro]

I want to investigate whether we should compute metrics only when doing loss.backward() using Lightning features.