TorchMetrics Not working in google colab

[bibinwils]

I have taken the torchmetrics for regression task in google colab. and it's taking so much time and GPU memory to run. So I removed them from the train and validation step and put them only in the test_step. But after the training, in test state also it's not completing the run, and it's keeping on rolling. What should I do?

[SkafteNicki]

Hi, could you please provide a link to your colab notebook or provide some code example for better understanding your problem.

[bibinwils]

Sorry, since I couldn't get any results I stopped in between. when I looked to get the results, still it's not possible to run when torchmetrics are enabled. This is the code: https://colab.research.google.com/drive/12-wApssxCngkWF9u1Rtcug556CGmtBUT3o?usp=sharing can you please look into it?