为什么PyTorch比滑雪板模型更快？

Question

最近，我了解了Python的蜂鸟库。我在一个有两个标签的10m大小的数据集上训练了一个RandomForest。用sklearn进行推断需要450 ms。但是，在将相同的模型转换为PyTorch后，现在需要128 on的CPU推理。

如果两者都运行在CPU上，那么为什么蜂鸟的Pytorch模型比sklean模型更快呢？

我不明白蜂鸟对我的滑雪板模型做了什么来提高速度。

Answer 1

如果没有访问代码的权限，很难回答您的问题。理解差异的最好方法是分析代码并查看针对特定问题的瓶颈所在。

为此，您可以在python中使用不同的分析模块：

1. cProfile
2. python行分析器