fixed conv2d_test

hycis · hycis · commit ee02a6d18310 · 2018-04-13T15:41:14.000+08:00
diff --git a/test/layer_conv_test.py b/test/layer_conv_test.py
@@ -50,7 +50,7 @@ def test_Atrous_Conv2d():
         sess.run(init)
         out = sess.run(y_sb, feed_dict={X_ph:np.random.rand(32, h, w, c)})
         print(out.shape)
-        assert out.shape[1] == h - 2*((k+(k-1)*(r-1))/2), out.shape[2] == w - 2*((w+(w-1)*(r-1))/2)
+        assert out.shape[1] == h - 2*int((k+(k-1)*(r-1))/2), out.shape[2] == w - 2*int((w+(w-1)*(r-1))/2)
 
 
 
