Paul Barbier

How to restore on a CPU a checkpoint saved on a GPU?

1

Hi, I made checkpoints of my weights and some Jax arrays using the following snippet on a GPU instance (v100): ``` import orbax.checkpointer as ocp checkpointer = ocp.PyTreeCheckpointer() checkpointer.save(path, checkpoint)...