Test if possible conf_mat from base_task into a callback

[lvoegtlin]

Is your feature request related to a problem? Please describe. The problem before with the conf mat callback was that it had a semaphore leak. As described here (https://github.com/ashleve/lightning-hydra-template/issues/189#issuecomment-1003532448), it should work now with the usage of torchmetrics.

Checklist

• [ ] Factor the conf mat log into callback
• [ ] Extensice testing
• [ ] Tests
• [ ] PR

[lvoegtlin]

Move it into Wandb directly to have more interactive features