Update resnet.py

Enable training for self.stem
ncepu-liudong · Sep 8, 2019 · 37da9da · 37da9da
1 parent 188b47b
commit 37da9da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/深度学习与TensorFlow入门实战-源码和PPT/lesson43-ResNet/resnet.py b/深度学习与TensorFlow入门实战-源码和PPT/lesson43-ResNet/resnet.py
@@ -69,7 +69,7 @@ def __init__(self, layer_dims, num_classes=100): # [2, 2, 2, 2]
 
     def call(self, inputs, training=None):
 
-        x = self.stem(inputs)
+        x = self.stem(inputs,training=training)
 
         x = self.layer1(x,training=training)
         x = self.layer2(x,training=training)