orbax

orbax copied to clipboard

Use thread-safe barrier api by default in Orbax.

copybara-service[bot]

Internal change.

copybara-service[bot]

internal changes only

1

internal changes only

copybara-service[bot]

Internal change.

1

Internal change.

copybara-service[bot]

Internal

1

Internal

copybara-service[bot]

Add an option to skip validating merged parameters in Orbax.

copybara-service[bot]

Internal

1

Internal

copybara-service[bot]

Warnings When Restoring the Params

1

The code shows like the following. It could run but prompted some warning: /opt/conda/lib/python3.10/site-packages/orbax/checkpoint/type_handlers.py:1544: UserWarning: Couldn't find sharding info under RestoreArgs. Populating sharding info from sharding file. Please note restoration...

richardmkit

Can't checkpoint Numpy RNG state

6

Hi. So I'm trying to use orbax-checkpoint to checkpoint my full experiment state, not just my network weights, and part of this state is NumPy RNG states, which look as...

rainx0r

Internal change.

copybara-service[bot]