From 7484c54f035f9e6f81e880004694670273675010 Mon Sep 17 00:00:00 2001
From: Dhairya Gandhi <dhairya@juliacopmuting.com>
Date: Tue, 8 Jan 2019 00:32:55 +0530
Subject: [PATCH] fix train! API syntax docstring

---
 src/optimise/train.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index 23c41373..571627a1 100644
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@@ -45,7 +45,7 @@ function stop()
 end
 
 """
-    train!(model, loss, data, opt)
+    train!(loss, params, data, opt; cb = () -> ())
 
 For each datapoint `d` in `data` computes the gradient of `loss(d...)` through
 backpropagation and calls the optimizer `opt`.
@@ -54,7 +54,7 @@ Takes a callback as keyword argument `cb`. For example, this will print "trainin
 every 10 seconds:
 
 ```julia
-Flux.train!(model, loss, data, opt,
+Flux.train!(loss, params, data, opt,
             cb = throttle(() -> println("training"), 10))
 ```