Skip to content

Commit

Permalink
Update SavingAndLoadingModelWeights.md
Browse files Browse the repository at this point in the history
  • Loading branch information
AqwamCreates authored Jan 27, 2025
1 parent f7a0565 commit 0ec2318
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/Tutorials/SavingAndLoadingModelWeights.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ DataPredict Neural provides the ability to save and load weights from trained mo

* Containers.

## Saving And Loading Weights From BaseWeightBlock Classes
## Saving And Loading Weights From The BaseWeightBlock Classes

In order to save the weights from BaseWeightBlock classes, we first need to call the getWeightTensor() function on one of our weight blocks.

Expand All @@ -26,7 +26,7 @@ Linear:setWeightTensor(savedWeightTensor)

```

## Saving And Loading Weights From Containers
## Saving And Loading Weights From The Containers

In order to save the weights from Containers, we first need to call the getWeightTensorArray() function.

Expand Down

0 comments on commit 0ec2318

Please sign in to comment.