From a27d727e643b5149ef00b247e2e144a05b0ab3ab Mon Sep 17 00:00:00 2001
From: Juncheng <liujuncheng1022@gmail.com>
Date: Mon, 2 Sep 2019 14:23:20 +0800
Subject: [PATCH] fix job_type (#2102)
---
oneflow/python/framework/config_util.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/oneflow/python/framework/config_util.py b/oneflow/python/framework/config_util.py
index 75ae38a34..f02f0dbb3 100644
--- a/oneflow/python/framework/config_util.py
+++ b/oneflow/python/framework/config_util.py
@@ -305,4 +305,5 @@ def _DefaultConfigCppFlags(config):
def _TryCompleteDefaultJobConfigProto(job_conf):
- job_conf.predict_conf.SetInParent()
+ if job_conf.WhichOneof('job_type') is None:
+ job_conf.predict_conf.SetInParent()
--
GitLab