diff --git a/example/imagenet/README.md b/example/imagenet/README.md index 1713c3886d3..5f23292ce16 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 1b90caa4281..e4e1663406c 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))