flax.config 包

Flax 的全局配置标志。

class flax.configurations.Config

property flax_filter_frames

是否从回溯中隐藏 flax 内部堆栈帧。

property flax_fix_rng_separator

是否在将静态数据折叠到 PRNG 密钥中时添加分隔符字符。

property flax_preserve_adopted_names

在采用外部模块时，不要覆盖现有名称。

property flax_profile

是否在 jax.named_scope 下运行模块方法以进行分析。

property flax_return_frozendict

是否在调用 init 或 apply 时返回 FrozenDicts。

property flax_use_orbax_checkpointing

是否使用 Orbax 保存检查点。

update(name_or_holder, value, /)

修改给定标志的值。

参数

• name_or_holder – 要修改的标志的名称或相应的标志持有者对象。

• value – 要设置的新值。