orbax
orbax copied to clipboard
NotImplementedError: CopyRange not supported
See this code sample:
import pathlib
import orbax.checkpoint
orbax_checkpointer = orbax.checkpoint.PyTreeCheckpointer()
tree = {'key': 7}
path = pathlib.Path(__file__).parent / 'foo'
orbax_checkpointer.save(path, tree)
When I run it on Linux, it works fine. When I run it on Windows 10, I get this traceback with the exception: builtins.NotImplementedError: CopyRange not supported
.
The versions I use:
orbax-checkpoint==0.5.23
tensorstore==0.1.64