We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
allDifferentiableVariables
1 parent 085217d commit 4cfaa4cCopy full SHA for 4cfaa4c
README.md
@@ -57,7 +57,7 @@ for _ in 0..<1000 {
57
print("Loss: \(loss)")
58
return loss
59
}
60
- optimizer.update(&classifier.allDifferentiableVariables, along: 𝛁model)
+ optimizer.update(&classifier, along: 𝛁model)
61
62
```
63
@@ -69,7 +69,7 @@ for _ in 0..<1000 {
69
let (loss, 𝛁ŷ) = ŷ.valueWithGradient { ŷ in softmaxCrossEntropy(logits: ŷ, labels: y) }
70
print("Model output: \(ŷ), Loss: \(loss)")
71
let 𝛁model = backprop(𝛁ŷ)
72
73
74
75
0 commit comments