训练代码的时候，cpu只有一个核心两个线程被调用，这是正常的吗？我的cpu是5950x，显卡是4090。

您好，在复现您的工作时，无论flatten-transformer还是agent-attention，我的电脑在复现的时候都存在这样的问题，请问您知道是什么原因吗？是否在哪里有分配cpu核心数的代码？