模块 'trax.supervised' 没有属性 'inputs'

问题描述

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

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

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

谢谢!

解决方法

根据 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