Update resnet.py

Enable training else the parameter will not get updated

Author: kelvinkoh0308

深度学习与TensorFlow入门实战-源码和PPT/lesson43-ResNet/resnet.py

@@ -28,11 +28,11 @@ def call(self, inputs, training=None):
 
         # [b, h, w, c]
         out = self.conv1(inputs)
-        out = self.bn1(out)
+        out = self.bn1(out,training=training)
         out = self.relu(out)
 
         out = self.conv2(out)
-        out = self.bn2(out)
+        out = self.bn2(out,training=training)
 
         identity = self.downsample(inputs)
 
@@ -71,10 +71,10 @@ def call(self, inputs, training=None):
 
         x = self.stem(inputs)
 
-        x = self.layer1(x)
-        x = self.layer2(x)
-        x = self.layer3(x)
-        x = self.layer4(x)
+        x = self.layer1(x,training=training)
+        x = self.layer2(x,training=training)
+        x = self.layer3(x,training=training)
+        x = self.layer4(x,training=training)
 
         # [b, c]
         x = self.avgpool(x)
@@ -102,4 +102,4 @@ def resnet18():
 
 
 def resnet34():
-    return ResNet([3, 4, 6, 3])
+    return ResNet([3, 4, 6, 3])