Skip to content

Commit 1730eed

Browse files
nkconnork-w-w
authored andcommitted
Bug fix on flag name in wide_deep.py (#3777)
PR #3460 renamed FLAGS.epochs_per_eval to flags.epochs_between_evals; it missed this one
1 parent 86cb0aa commit 1730eed

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

official/wide_deep/wide_deep.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -182,9 +182,9 @@ def main(argv):
182182
train_file = os.path.join(flags.data_dir, 'adult.data')
183183
test_file = os.path.join(flags.data_dir, 'adult.test')
184184

185-
# Train and evaluate the model every `FLAGS.epochs_per_eval` epochs.
185+
# Train and evaluate the model every `flags.epochs_between_evals` epochs.
186186
def train_input_fn():
187-
return input_fn(train_file, flags.epochs_per_eval, True, flags.batch_size)
187+
return input_fn(train_file, flags.epochs_between_evals, True, flags.batch_size)
188188

189189
def eval_input_fn():
190190
return input_fn(test_file, 1, False, flags.batch_size)
@@ -194,7 +194,7 @@ def eval_input_fn():
194194
tensors_to_log={'average_loss': 'head/truediv',
195195
'loss': 'head/weighted_loss/Sum'})
196196

197-
# Train and evaluate the model every `FLAGS.epochs_between_evals` epochs.
197+
# Train and evaluate the model every `flags.epochs_between_evals` epochs.
198198
for n in range(flags.train_epochs // flags.epochs_between_evals):
199199
model.train(input_fn=train_input_fn, hooks=train_hooks)
200200
results = model.evaluate(input_fn=eval_input_fn)

0 commit comments

Comments
 (0)