From f6907bea3bcfb51798863f7f04802a4b5abebe60 Mon Sep 17 00:00:00 2001 From: Enzo L F <49812445+lfenzo@users.noreply.github.com> Date: Sat, 4 Jun 2022 14:42:01 -0300 Subject: [PATCH] docs: Update docs/src/gpu.md (common gpu workflows) Co-authored-by: Brian Chen --- docs/src/gpu.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/src/gpu.md b/docs/src/gpu.md index da40723abb..cced9ca993 100644 --- a/docs/src/gpu.md +++ b/docs/src/gpu.md @@ -137,7 +137,7 @@ In order to train the model using the GPU both model and the training data have ### Saving GPU-Trained Models -After the training process is done one must always transfer the trained model back to the `cpu` memory scope before saving it to secundary memory. This can be done, as described in the previous section, with: +After the training process is done, one must always transfer the trained model back to the `cpu` memory scope before serializing or saving to disk. This can be done, as described in the previous section, with: ```julia model = cpu(model) # or model = model |> cpu ```