在colab运行gpt微调时遇到错误 RuntimeError: Index tensor must have the same number of dimensions as self tensor

[AmemiaErika]

[rank0]:   File "/usr/local/lib/python3.9/site-packages/torchmetrics/functional/classification/stat_scores.py", line 364, in _refine_preds_oh
[rank0]:     return torch.zeros_like(preds_oh, dtype=torch.int32).scatter_(-1, result.unsqueeze(1).unsqueeze(1), 1)
[rank0]: RuntimeError: Index tensor must have the same number of dimensions as self tensor

[yangjianxie]

我也遇到了这个问题

[yangjianxie]

我刚遇到这个问题...试了一下解决了。
方法是：
pip install torchmetrics==1.5

从另一个issue中看到的解决方法，我试了下可以