From b8d2420527bca5d5d86d34d5161c1681d4ad7898 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Marci=C5=84czuk?= <marcinczuk@gmail.com>
Date: Mon, 24 Oct 2022 11:25:10 +0200
Subject: [PATCH] Pass list of paths.
---
train-batch.py | 13 ++++++++-----
1 file changed, 8 insertions(+), 5 deletions(-)
diff --git a/train-batch.py b/train-batch.py
index 863870f..4cee478 100644
--- a/train-batch.py
+++ b/train-batch.py
@@ -22,7 +22,12 @@ def load_configurations(path: str) -> Dict[str, Union[str, List[str]]]:
def generate_cmds_list(params: Dict[str, Union[str, List[str]]]) -> List[List[str]]:
lists = [[]]
for key, values in params.items():
- if isinstance(values, list):
+ if key.endswith("/list"):
+ key = key[:-5]
+ for ll in lists:
+ for value in values:
+ ll.extend([f"--{key}", value])
+ elif isinstance(values, list):
new_lists = []
for value in values:
for ll in lists:
@@ -33,6 +38,7 @@ def generate_cmds_list(params: Dict[str, Union[str, List[str]]]) -> List[List[st
ll.append(f"--{key}")
if values is not None:
ll.append(str(values))
+ print(lists)
return lists
@@ -57,7 +63,4 @@ if __name__ == '__main__':
for cmd in cmds:
args = parser.parse_args(cmd)
print(args)
- try:
- train_model(args)
- except KeyError as error:
- print(error)
+ train_model(args)
--
GitLab