Forward current environment variables when autotuning batch size (#849)

OpenNMT · Jul 1, 2021 · 9494332 · 9494332
1 parent 1045399
commit 9494332
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/opennmt/runner.py b/opennmt/runner.py
@@ -559,9 +559,8 @@ def _auto_tune_batch_size(
             with tf.io.gfile.GFile(config_path, mode="w") as config_file:
                 yaml.dump(run_config, config_file)
 
-            env = {
-                "TF_CPP_MIN_LOG_LEVEL": "2",
-            }
+            env = os.environ.copy()
+            env["TF_CPP_MIN_LOG_LEVEL"] = "2"
             args = [
                 sys.executable or "python",
                 "-m",