Skip to content

Commit

Permalink
typo
Browse files Browse the repository at this point in the history
  • Loading branch information
bchess committed Apr 30, 2024
1 parent 3a139c0 commit 731f920
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/subprocess-serialization.md
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ if __name__ == '__main__':
Once the tensors are in CPU memory, they no longer need to occupy CUDA memory. But the tensors
will now need to occupy CPU memory until they are fully serialized.

Do this by calling `model.to("cpu")` immediately after sending to serializer.
Do this by calling `model.to("cpu")` immediately after sending to the serializer.

If you like, you can also use some sort of IPC object to communicate back to the
host process when the snapshotting has finished so you know when the CUDA memory
Expand Down

0 comments on commit 731f920

Please sign in to comment.