module - 模块“trax.supervised”没有属性“输入”

Question

我正在做一个涉及 Google 的 TRAX 包的项目。在 Jupyter notebook 上运行一段代码时，我遇到了以下错误消息。

module 'trax.supervised' has no attribute 'inputs'

有没有办法解决这个问题？我有 trax 版本 1.2.3。

谢谢！

Answer 1

根据 trax 文档调用必须是 trax.data 而不是 trax.Supervised trax.data.inputs.add_loss_weights(generator, id_to_mask=None)

https://trax-ml.readthedocs.io/en/latest/trax.data.html