From 6757f905533be8bd13951691400bc9c02dfd9918 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9s=20Marafioti?= <andimarafioti@gmail.com>
Date: Wed, 18 Sep 2024 16:51:17 +0200
Subject: [PATCH] Apply suggestions from code review
Co-authored-by: eustlb <94853470+eustlb@users.noreply.github.com>
---
s2s_pipeline.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/s2s_pipeline.py b/s2s_pipeline.py
index bd91fcd..fc3d433 100644
--- a/s2s_pipeline.py
+++ b/s2s_pipeline.py
@@ -85,8 +85,10 @@ def parse_arguments():
)
if len(sys.argv) == 2 and sys.argv[1].endswith(".json"):
+ # Parse configurations from a JSON file if specified
return parser.parse_json_file(json_file=os.path.abspath(sys.argv[1]))
else:
+ # Parse arguments from command line if no JSON file is provided
return parser.parse_args_into_dataclasses()
@@ -98,6 +100,7 @@ def setup_logger(log_level):
)
logger = logging.getLogger(__name__)
+ # torch compile logs
if log_level == "debug":
torch._logging.set_logs(graph_breaks=True, recompiles=True, cudagraphs=True)
--
GitLab