diff --git a/papers/deep_residual_learning/Deep_Residual_Learning_CIFAR-10.py b/papers/deep_residual_learning/Deep_Residual_Learning_CIFAR-10.py old mode 100644 new mode 100755 index 939685a..6bd4f6d --- a/papers/deep_residual_learning/Deep_Residual_Learning_CIFAR-10.py +++ b/papers/deep_residual_learning/Deep_Residual_Learning_CIFAR-10.py @@ -113,7 +113,7 @@ def residual_block(l, increase_dim=False, projection=False): block = NonlinearityLayer(ElemwiseSumLayer([stack_2, projection]),nonlinearity=rectify) else: # identity shortcut, as option A in paper - identity = ExpressionLayer(l, lambda X: X[:, :, ::2, ::2], lambda s: (s[0], s[1], s[2]//2, s[3]//2)) + identity = ExpressionLayer(l, lambda X: X[:, :, ::2, ::2], lambda s: (s[0], s[1], (s[2]+1)//2, (s[3]+1)//2)) padding = PadLayer(identity, [out_num_filters//4,0,0], batch_ndim=1) block = NonlinearityLayer(ElemwiseSumLayer([stack_2, padding]),nonlinearity=rectify) else: