From f5f5e772395c843bc39d4aa05a8015cb1040b680 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 27 Apr 2017 15:04:48 +0100 Subject: [PATCH] fix second unsqueeze --- examples/MNIST.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/MNIST.jl b/examples/MNIST.jl index fa3576af..7befebf8 100644 --- a/examples/MNIST.jl +++ b/examples/MNIST.jl @@ -19,4 +19,4 @@ model(unsqueeze(data[1][1])) Flux.train!(model, train, test, η = 1e-4) # An example prediction post-training -model(data[1][1]) +model(unsqueeze(data[1][1]))