Python 用于文本分类多类的包装器Ktrain-使用CPU并行化

我正试图利用Ktrain包装器解决一个与BERT有关的多类分类问题。由于有大量数据，我想知道是否可以并行化learner.fit_onecycle（3e-4,5）函数或autofit（）函数

例如，在sklerarn中，使用n_jobs参数指定并行作业的数量

你能帮我吗？

---

首先谢谢你，如果你在训练中使用DistilBert而不是BERT，你的训练时间会减少一半。第二，默认情况下，TensorFlow2在使用CPU调用ktrain中的任何fit方法时都应使用多核。你可以，如果你愿意的话