From e4c2be618474c9eef2d716c57a76d2d0d78662ef Mon Sep 17 00:00:00 2001
From: Richard Wei <rxwei@google.com>
Date: Sat, 23 Feb 2019 19:33:39 -0800
Subject: [PATCH] Update README sample code to reflect #28.

Should go in after #28 gets merged.
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 035827992..ab1dc9b32 100644
--- a/README.md
+++ b/README.md
@@ -40,8 +40,8 @@ struct Model: Layer {
 #### Initialize a model and an optimizer
 
 ```swift
-let optimizer = SGD<Model, Float>(learningRate: 0.02)
 var classifier = Model()
+let optimizer = SGD(learningRate: 0.02, for: classifier)
 let context = Context(learningPhase: .training)
 let x: Tensor<Float> = ...
 let y: Tensor<Float> = ...