Update resnet.py

enable training for self.stem
This commit is contained in:
kelvinkoh0308
2019-09-11 14:43:57 +09:00
committed by GitHub
parent dbd0334397
commit 6b796479e6

View File

@@ -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(inputstraining=training)
out = tf.nn.relu(out)
# print('stem:',out.shape)
@@ -116,4 +116,4 @@ def main():
if __name__ == '__main__':
main()
main()