From d9df55622efb969780713531c81a1d91c7ca3f3e Mon Sep 17 00:00:00 2001
From: Bing Xu <antinucleon@gmail.com>
Date: Mon, 21 Sep 2015 16:31:16 -0600
Subject: [PATCH] [BUG FIX] Slice in float type

---
 example/cifar10/cifar10.py                     | 18 +++++++++---------
 example/{imagenet => notebooks}/alexnet.ipynb  |  0
 .../composite_symbol.ipynb                     |  0
 python/mxnet/model.py                          |  4 ++--
 4 files changed, 11 insertions(+), 11 deletions(-)
 rename example/{imagenet => notebooks}/alexnet.ipynb (100%)
 rename example/{python-howto => notebooks}/composite_symbol.ipynb (100%)

diff --git a/example/cifar10/cifar10.py b/example/cifar10/cifar10.py
index b99c49ea7423..bf32308846e1 100644
--- a/example/cifar10/cifar10.py
+++ b/example/cifar10/cifar10.py
@@ -5,6 +5,7 @@
 sys.path.insert(0, "../../python/")
 sys.path.append("../../tests/python/common")
 # import library
+import logging
 import mxnet as mx
 import get_data
 import time
@@ -59,10 +60,6 @@
 [39]  train-error:0.00125879  val-error:0.0833
 [40]  train-error:0.000699329 val-error:0.0842
 """
-def CalAcc(out, label):
-    pred = np.argmax(out, axis=1)
-    return np.sum(pred == label) * 1.0 / out.shape[0]
-
 
 np.random.seed(1812)
 
@@ -178,11 +175,14 @@ def RandomInit(narray):
         preprocess_threads=1)
 
 def test_cifar():
-    model = mx.model.MXNetModel(ctx=mx.gpu(),
-            symbol=loss, data=(batch_size, 3, 28, 28),
-            optimizer="sgd", num_round = epoch, batch_size = batch_size,
-            learning_rate=0.05, momentum=0.9, weight_decay=0.00001)
-    model.fit(X=train_dataiter, eval_set=test_dataiter, eval_metric=CalAcc)
+    logging.basicConfig(level=logging.DEBUG)
+    console = logging.StreamHandler()
+    console.setLevel(logging.DEBUG)
+    logging.getLogger('').addHandler(console)
+    # get model from symbol
+    model = mx.model.FeedForward(ctx=mx.gpu(), symbol=loss, num_round = epoch,
+                                 learning_rate=0.05, momentum=0.9, wd=0.00001)
+    model.fit(X=train_dataiter, eval_data=test_dataiter)
 
 
 if __name__ == "__main__":
diff --git a/example/imagenet/alexnet.ipynb b/example/notebooks/alexnet.ipynb
similarity index 100%
rename from example/imagenet/alexnet.ipynb
rename to example/notebooks/alexnet.ipynb
diff --git a/example/python-howto/composite_symbol.ipynb b/example/notebooks/composite_symbol.ipynb
similarity index 100%
rename from example/python-howto/composite_symbol.ipynb
rename to example/notebooks/composite_symbol.ipynb
diff --git a/python/mxnet/model.py b/python/mxnet/model.py
index 1fd73eb4033e..11e440988f51 100644
--- a/python/mxnet/model.py
+++ b/python/mxnet/model.py
@@ -104,8 +104,8 @@ def _split_input_slice(input_shape, num_split):
     slices = []
     shapes = []
     for k in range(num_split):
-        begin = min(k * step, batch_size)
-        end = min((k+1) * step, batch_size)
+        begin = int(min(k * step, batch_size))
+        end = int(min((k+1) * step, batch_size))
         if begin == end:
             raise ValueError('Too many slices such that some splits are empty')
         slices.append(slice(begin, end))