Make train section optional for inference only configurations
@@ -237,6 +237,7 @@ def main():
model_dir=config["model_dir"],
session_config=session_config)
+ if "train" in config:
if "save_summary_steps" in config["train"]:
run_config = run_config.replace(
save_summary_steps=config["train"]["save_summary_steps"],
