mirror of
https://github.com/dragen1860/TensorFlow-2.x-Tutorials.git
synced 2021-05-12 18:32:23 +03:00
Update resnet.py
enable training for self.stem
This commit is contained in:
@@ -81,7 +81,7 @@ class ResNet(keras.Model):
|
||||
|
||||
def call(self, inputs, training=None):
|
||||
# print('x:',inputs.shape)
|
||||
out = self.stem(inputs)
|
||||
out = self.stem(inputs,training=training)
|
||||
out = tf.nn.relu(out)
|
||||
|
||||
# print('stem:',out.shape)
|
||||
@@ -116,4 +116,4 @@ def main():
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user