导入 trax 加载时间过长
Import trax takes too long to load
我第一次加载这个库时被难住了。在我的本地计算机上，我至少需要 40 秒才能在本地 Jupyter Notebook 上加载 trax，在共享的 Colab 环境中加载它需要 1 多分钟。
import trax
我不确定这是我的安装问题还是我使用的 trax 版本中的 BUG？
我是 trax 的新手，事实上我的经验是使用 Keras 和 TensorFlow，所以我想听听 trax 社区的人的意见，看看这是否正常。
提前致谢！
顺便说一句：我正在使用 trax 1.4.1 和 Python 3.9.6，我的本地计算机具有以下规格：
Intel(R) Core(TM) i5-6200U CPU @ 2.30GHz, 4 cores and 16GB RAM.
经过一段时间的搜索，这似乎是正常现象。事实上，官方 Trax 存储库中提出了一个问题：import trax takes 17 seconds #1368.
显然，包含用于数学运算的 trax re-implements 的 fastmath 模块有很多依赖项。
来自问题线程:
This is a well known problem occurring on basically all setups (local, colab, gpu cluster) and it is not a big issue for running long experiments, however it does make local debugging hard. I have tried debugging the import graph with profiler, but without success yet. It looks like even from trax import fastmath has plenty of dependencies - here is the tree generated by importlab library for trax.fastmath.init module:
importlab --tree init.py
out:
https://gist.github.com/syzymon/3bb6f59063f918b4b62b77cdb223da72
所以总而言之，不管你需要 17 秒还是
像我一样 40 秒是 Trax 中的已知行为。
我第一次加载这个库时被难住了。在我的本地计算机上，我至少需要 40 秒才能在本地 Jupyter Notebook 上加载 trax，在共享的 Colab 环境中加载它需要 1 多分钟。
import trax
我不确定这是我的安装问题还是我使用的 trax 版本中的 BUG？
我是 trax 的新手，事实上我的经验是使用 Keras 和 TensorFlow，所以我想听听 trax 社区的人的意见，看看这是否正常。
提前致谢！
顺便说一句：我正在使用 trax 1.4.1 和 Python 3.9.6，我的本地计算机具有以下规格：
Intel(R) Core(TM) i5-6200U CPU @ 2.30GHz, 4 cores and 16GB RAM.
经过一段时间的搜索，这似乎是正常现象。事实上，官方 Trax 存储库中提出了一个问题：import trax takes 17 seconds #1368.
显然，包含用于数学运算的 trax re-implements 的 fastmath 模块有很多依赖项。
来自问题线程:
This is a well known problem occurring on basically all setups (local, colab, gpu cluster) and it is not a big issue for running long experiments, however it does make local debugging hard. I have tried debugging the import graph with profiler, but without success yet. It looks like even from trax import fastmath has plenty of dependencies - here is the tree generated by importlab library for trax.fastmath.init module: importlab --tree init.py out: https://gist.github.com/syzymon/3bb6f59063f918b4b62b77cdb223da72
所以总而言之，不管你需要 17 秒还是 像我一样 40 秒是 Trax 中的已知行为。