Is your feature request related to a problem? Please describe.
Currently GaNDLF is using deprecated version of tochmetrics, making the migration to torch 2.0 impossible.

Describe the solution you'd like
Modify the API for compatibility with torchmetrics 1.1.2.

Describe alternatives you've considered

Additional context