Commit 5610f5d
committed
Fix a logic of Weight::applyGradient in nntrainer/tesnor/weight.cpp
This commit fixes a logic in `Weight::applyGradient(double lr, Tensor &updated_grad)` in `nntrainer/tensor/weight.cpp`
Previous version fallbacks to `applyGradient(lr)`, which does not use `updated_grad`.
This version now use `updated_grad` to apply gradients for FP32 path
Signed-off-by: PJH6029 <[email protected]>1 parent 2470bbc commit 5610f5d
1 file changed
+2
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
137 | 137 | | |
138 | 138 | | |
139 | 139 | | |
140 | | - | |
| 140 | + | |
| 141 | + | |
141 | 142 | | |
142 | 143 | | |
143 | 144 | | |
| |||
0 commit comments