Fixed bug normalizer

parent 7ed37527
......@@ -25,9 +25,8 @@ class MeanOffset(object):
self.mean_offset = mean_offset
def __call__(self, x):
for i in range(len(self.mean_offset)):
x[:, i, :, :] = x[:, i, :, :] - self.mean_offset[i]
x[:, :, i] = x[:, :, i] - self.mean_offset[i]
return x
......
......@@ -53,13 +53,13 @@ class VGG16_mod(SequenceNetwork):
# Forth convolutional block
conv5_1_kernel_size=3,
conv5_1_output=512,
conv5_1_output=256,
conv5_2_kernel_size=3,
conv5_2_output=512,
conv5_2_output=256,
conv5_3_kernel_size=3,
conv5_3_output=512,
conv5_3_output=256,
n_classes=10,
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment