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 – 要设置的新值。