Skip to content

Commit 4cfaa4c

Browse files
authored
Remove allDifferentiableVariables usage from README. (tensorflow#420)
1 parent 085217d commit 4cfaa4c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ for _ in 0..<1000 {
5757
print("Loss: \(loss)")
5858
return loss
5959
}
60-
optimizer.update(&classifier.allDifferentiableVariables, along: 𝛁model)
60+
optimizer.update(&classifier, along: 𝛁model)
6161
}
6262
```
6363

@@ -69,7 +69,7 @@ for _ in 0..<1000 {
6969
let (loss, 𝛁ŷ) = ŷ.valueWithGradient { ŷ in softmaxCrossEntropy(logits: ŷ, labels: y) }
7070
print("Model output: \(ŷ), Loss: \(loss)")
7171
let 𝛁model = backprop(𝛁ŷ)
72-
optimizer.update(&classifier.allDifferentiableVariables, along: 𝛁model)
72+
optimizer.update(&classifier, along: 𝛁model)
7373
}
7474
```
7575

0 commit comments

Comments
 (0)