Skip to content

Commit 4545fb1

Browse files
authored
Merge pull request tensorflow#2744 from tensorflow/wide-deep-train-longer
Train wide_deep.py for longer for more consistent results that match the tutorials
2 parents fd5e351 + db0de3b commit 4545fb1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

official/wide_deep/wide_deep.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@
4444
help="Valid model types: {'wide', 'deep', 'wide_deep'}.")
4545

4646
parser.add_argument(
47-
'--train_epochs', type=int, default=20, help='Number of training epochs.')
47+
'--train_epochs', type=int, default=40, help='Number of training epochs.')
4848

4949
parser.add_argument(
5050
'--epochs_per_eval', type=int, default=2,
@@ -213,7 +213,7 @@ def main(unused_argv):
213213

214214
# Display evaluation metrics
215215
print('Results at epoch', (n + 1) * FLAGS.epochs_per_eval)
216-
print('-' * 30)
216+
print('-' * 60)
217217

218218
for key in sorted(results):
219219
print('%s: %s' % (key, results[key]))

0 commit comments

Comments
 (0)