alihan/TensorFlow-2.x-Tutorials
1
0
Fork 0
mirror of https://github.com/dragen1860/TensorFlow-2.x-Tutorials.git synced 2021-05-12 18:32:23 +03:00
Code Issues Projects Releases Wiki Activity

Update resnet.py

Browse Source 
Enable training for self.stem
This commit is contained in:
kelvinkoh0308
2019-09-08 14:41:18 +09:00
committed by GitHub
parent 188b47b717
commit 37da9da39b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -69,7 +69,7 @@ class ResNet(keras.Model):
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)