TorchMetrics Not working in google colab #562
Unanswered
bibinwils
asked this question in
Regression
Replies: 1 comment 1 reply
-
|
Hi, could you please provide a link to your colab notebook or provide some code example for better understanding your problem. |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
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? |
Beta Was this translation helpful? Give feedback.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
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?
Beta Was this translation helpful? Give feedback.
All reactions