Merge pull request #147 from mli/master

[example] minor fix for alexnet
apache · Sep 24, 2015 · 30d5ecd · 30d5ecd
2 parents f326831 + 52ff3b6
commit 30d5ecd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git 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
@@ -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))