Skip to content

Commit

Permalink
docs: Update docs/src/gpu.md (common gpu workflows)
Browse files Browse the repository at this point in the history
Co-authored-by: Brian Chen <[email protected]>
  • Loading branch information
lfenzo and ToucheSir authored Jun 4, 2022
1 parent 3ba5718 commit f6907be
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/src/gpu.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
```
Expand Down

0 comments on commit f6907be

Please sign in to comment.