Update 20200117-tfx-generic-trainer.md

Author: 1025KB

rfcs/20200117-tfx-generic-trainer.md

@@ -237,8 +237,10 @@ def run_fn(args: trainer.executor.TrainerFnArgs) -> None:
   """Build the TF model and train it."""
   ps_strategy = tf.distribute.experimental.ParameterServerStrategy()
   with ps_strategy.scope():
+    train_dataset = _create_dataset(args.train_files, ...)
+    eval_dataset = _create_dataset(args.eval_files, ...)
     model = _build_keras_model()
-  model.fit(...)
+  model.fit(train_dataset, validation_data=eval_dataset, ...)
   if _is_chief():
     model.save(...)
 ```