orbax

orbax copied to clipboard

Package(s) vendoring issues

4

What is supposed way to distribute `orbax-checkpoint` и `orbax-export`? It is totally unclear for me 1. how to get notification about new releases? 2. what is 'ground truth' package(s) sources:...

daskol

haiku dicts contain ~ in the keys which currently is not handled by orbax-export. I created a PR here #419

ethanluoyc

Support TensorSpecWithDefault in input_signature

copybara-service[bot]

Internal change.

copybara-service[bot]

If you copy checkpoints from HOME to gcs they can get deleted

3

because of this line https://github.com/google/orbax/blob/c7a7fd48ff094ac4167d3fb1c10f8b6c8de32b3a/checkpoint/orbax/checkpoint/utils.py#L669 They don't have success file but are in GCS so orbax thinks its tmp and cleans it up. I would suggest always or never saving...

sshleifer

Is it possible to store checkpoints in an external storage such as S3?

4

I wasn't able to find the answers to my questions in the docs, so I'll just ask here: * What storage types other than local filesystem are supported with orbax?...

hr0nix

Bumps [ipython](https://github.com/ipython/ipython) from 7.16.3 to 8.10.0. Release notes Sourced from ipython's releases. See https://pypi.org/project/ipython/ We do not use GitHub release anymore. Please see PyPI https://pypi.org/project/ipython/ Commits 15ea1ed release 8.10.0 560ad10...

dependabot[bot]

Make Orbax auto-publish conditional on build passing with prod dependencies.

copybara-service[bot]

I believe restore_args must be equal to {'restore_args': restore_args} instead. Otherwise this will fail with unexpected keyword argument error.

mehdiataei

Added ```checkpoint_steps``` parameter to checkpoints_config

1

Added ```checkpoint_steps``` parameter to checkpoints_config Description of parameter: able to insert additional config (like period) that you pass in a list of training steps, and we save checkpoints at that...

copybara-service[bot]