flax.config 套件

Flax 的全域設定旗標。

類別 flax.configurations.Config

屬性 flax_filter_frames

是否從追蹤中隱藏 linho內部堆疊框架。

屬性 flax_fix_rng_separator

將靜態資料串入 PRNG 金鑰時，是否加入分隔符號。

屬性 flax_preserve_adopted_names

採用外部模組時，不要覆蓋現有的名稱。

屬性 flax_profile

是否在特徵文件中使用 `jax.named_scope` 來執行模組方法。

屬性 flax_return_frozendict

呼叫 `init` 或 `apply` 時是否回傳 `FrozenDicts`。

屬性 flax_use_orbax_checkpointing

是否使用 Orbax 來儲存檢查點。

更新(name_or_holder, value, /)

變更指定旗標的值。

參數

• name_or_holder – 要變更的旗標名稱或對應的旗標持有者物件。

• value – 要設定的新值。