【RuntimeError: unique_by_key: failed to synchronize】问题解决

原创于 2024-03-22 18:05:04 发布 · 895 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#深度学习 #人工智能

收录于

学习专栏收录该内容

9 篇文章

订阅专栏

文章讲述了在使用PyTorch进行训练时遇到`RuntimeError:unique_by_key:failedtosynchronize`错误，原因可能是config文件中设置的类别数目与实际数据类别不符。作者提醒读者检查这两个数值的一致性，以解决此问题。

pytorch训练时，提示错误：RuntimeError: unique_by_key: failed to synchronize: cudaErrorAssert: device-side assert triggered
完整log如下：

Traceback (most recent call last):
  File "main.py", line 106, in <module>
    train()
  File "main.py", line 61, in train
    loss.backward()
  File "/data/miniconda3/envs/bert_chinese/lib/python3.8/site-packages/torch/_tensor.py", line 487, in backward
    torch.autograd.backward(
  File "/data/miniconda3/envs/bert_chinese/lib/python3.8/site-packages/torch/autograd/__init__.py", line 200, in backward
    Variable._execution_engine.run_backward(  # Calls into the C++ engine to run the backward pass
RuntimeError: unique_by_key: failed to synchronize: cudaErrorAssert: device-side assert triggered

可以看看config文件中设置的类别数目与数据中实际的类别数目是否一致。问题比较基础，但是报错很多同学不知道是哪里触发的，所以记录一下