diff --git a/examples/mnist_net2net.py b/examples/mnist_net2net.py index c8c05bff393..77a16b6a0a8 100644 --- a/examples/mnist_net2net.py +++ b/examples/mnist_net2net.py @@ -208,7 +208,7 @@ def deeper2net_conv2d(teacher_w): kh, kw, num_channel, filters = teacher_w.shape student_w = np.zeros_like(teacher_w) for i in range(filters): - student_w[(kh - 1) / 2, (kw - 1) / 2, i, i] = 1. + student_w[(kh - 1) // 2, (kw - 1) // 2, i, i] = 1. student_b = np.zeros(filters) return student_w, student_b