Fixed bug normalizer

parent 7ed37527
...@@ -25,9 +25,8 @@ class MeanOffset(object): ...@@ -25,9 +25,8 @@ class MeanOffset(object):
self.mean_offset = mean_offset self.mean_offset = mean_offset
def __call__(self, x): def __call__(self, x):
for i in range(len(self.mean_offset)): 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 return x
......
...@@ -53,13 +53,13 @@ class VGG16_mod(SequenceNetwork): ...@@ -53,13 +53,13 @@ class VGG16_mod(SequenceNetwork):
# Forth convolutional block # Forth convolutional block
conv5_1_kernel_size=3, conv5_1_kernel_size=3,
conv5_1_output=512, conv5_1_output=256,
conv5_2_kernel_size=3, conv5_2_kernel_size=3,
conv5_2_output=512, conv5_2_output=256,
conv5_3_kernel_size=3, conv5_3_kernel_size=3,
conv5_3_output=512, conv5_3_output=256,
n_classes=10, 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