updata path

7d55e846 · 赵威 · ed7d8cf8 · 7d55e846
Commit 7d55e846 authored Jul 28, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

main.py src/main.py +2 -2

No files found.
--- a/src/main.py
+++ b/src/main.py
@@ -40,8 +40,8 @@ def main():
    all_features = build_features(df)
    params = {"feature_columns": all_features, "hidden_units": [64, 32], "learning_rate": 0.1}
    model_path = str(Path("~/data/model_tmp/").expanduser())
-    if os.path.exists(model_path):
+    # if os.path.exists(model_path):
-        shutil.rmtree(model_path)
+    #     shutil.rmtree(model_path)
    session_config = tf.compat.v1.ConfigProto()
    session_config.gpu_options.allow_growth = True