Update 20200117-tfx-generic-trainer.md

Author: 1025KB

rfcs/20200117-tfx-generic-trainer.md

@@ -145,6 +145,7 @@ def run_fn(args: trainer.executor.TrainerFnArgs) -> None:
   """Build the TF model and train it."""
   model = _build_keras_model()
   model.fit(...)
+  # Save model to args.serving_model_dir.
   model.save(...)
 ```