From 52ff3b6357edfd7d5d915a4bf3afe19b106a07c5 Mon Sep 17 00:00:00 2001 From: Mu Li Date: Wed, 23 Sep 2015 22:28:17 -0400 Subject: [PATCH] [example] minor fix for alexnet --- example/imagenet/README.md | 8 ++++---- example/imagenet/alexnet.py | 5 +++-- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/example/imagenet/README.md b/example/imagenet/README.md index 1713c3886d3e..5f23292ce167 100644 --- a/example/imagenet/README.md +++ b/example/imagenet/README.md @@ -11,9 +11,9 @@ TODO ## Results -Machine: Dual Xeon E5-2680 2.8GHz, Dual GTX 980, Ubuntu 14.0, GCC 4.8, MKL, CUDA +Machine: Dual Xeon E5-2680 2.8GHz, GTX 980, Ubuntu 14.0, GCC 4.8, MKL, CUDA 7, CUDNN v3 -| | val accuracy | 1 x GTX 980 | 2 x GTX 980 | -| --- | ---: | ---: | ---: | ---: | -| `alexnet.py` | ? | ? | 400 img/sec | +| | val accuracy | 1 x GTX 980 | 2 x GTX 980 | 4 x GTX 980 | +| --- | ---: | ---: | ---: | ---: | ---: | +| `alexnet.py` | ? | ? | 1020 img/sec | | diff --git a/example/imagenet/alexnet.py b/example/imagenet/alexnet.py index 1b90caa42815..e4e1663406c4 100644 --- a/example/imagenet/alexnet.py +++ b/example/imagenet/alexnet.py @@ -44,7 +44,8 @@ ## data -train, val = ilsvrc12_iterator(batch_size=256, input_shape=(3,224,224)) +batch_size = 256 +train, val = ilsvrc12_iterator(batch_size=batch_size, input_shape=(3,224,224)) ## train num_gpus = 2 @@ -58,4 +59,4 @@ wd = 0.00001) logging.basicConfig(level = logging.DEBUG) model.fit(X = train, eval_data = val, - epoch_end_callback = mx.callback.Speedometer(100)) + epoch_end_callback = mx.callback.Speedometer(batch_size=batch_size))