Skip to content

Commit 31b9ef0

Browse files
committed
Formatting fixes
1 parent d3908eb commit 31b9ef0

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

guides/md/orbax_checkpoint.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,10 +52,12 @@ import orbax.checkpoint as ocp
5252
## Orbax Callback
5353

5454
We need to create two main utilities to manage Orbax checkpointing in Keras:
55+
5556
1. `KerasOrbaxCheckpointManager`: A wrapper around
5657
`orbax.checkpoint.CheckpointManager` for Keras models.
5758
`KerasOrbaxCheckpointManager` uses `Model`'s `get_state_tree` and
5859
`set_state_tree` APIs to save and restore the model variables.
60+
5961
2. `OrbaxCheckpointCallback`: A Keras callback that uses
6062
`KerasOrbaxCheckpointManager` to automatically save and restore model states
6163
during training.
@@ -278,8 +280,5 @@ d manifest.ocdbt
278280
08372fc5734e445753b38235cb522988 c8af54d085d2d516444bd71f32a3787c
279281
4601db15b67650f7c8818bfc8afeb9f5 cfe1e3ea313d637df6f6d2b2c66ca17a
280282
a6ca20e04d8fe161ed95f6f71e8fe113
281-
282-
/usr/local/google/home/ssaadat/anaconda3/envs/keras-io/lib/python3.10/pty.py:89: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.
283-
pid, fd = os.forkpty()
284283
```
285284
</div>

guides/orbax_checkpoint.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,10 +49,12 @@
4949
## Orbax Callback
5050
5151
We need to create two main utilities to manage Orbax checkpointing in Keras:
52+
5253
1. `KerasOrbaxCheckpointManager`: A wrapper around
5354
`orbax.checkpoint.CheckpointManager` for Keras models.
5455
`KerasOrbaxCheckpointManager` uses `Model`'s `get_state_tree` and
5556
`set_state_tree` APIs to save and restore the model variables.
57+
5658
2. `OrbaxCheckpointCallback`: A Keras callback that uses
5759
`KerasOrbaxCheckpointManager` to automatically save and restore model states
5860
during training.

0 commit comments

Comments
 (0)