blob: eb872d789a2294211d6489f75d54a121a86f8f8b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
"""Run a experiment from a config file."""
import json
from subprocess import check_call
import click
from loguru import logger
import yaml
def run_experiments(experiments_filename: str) -> None:
"""Run experiment from file."""
with open(experiments_filename) as f:
experiments_config = yaml.safe_load(f)
num_experiments = len(experiments_config["experiments"])
for index in range(num_experiments):
experiment_config = experiments_config["experiments"][index]
experiment_config["experiment_group"] = experiments_config["experiment_group"]
# cmd = f"python training/run_experiment.py --gpu=-1 '{json.dumps(experiment_config)}'"
cmd = f"poetry run run-experiment --gpu=-1 --save --experiment_config '{json.dumps(experiment_config)}'"
print(cmd)
check_call(cmd, shell=True)
@click.command()
@click.option(
"--experiments_filename",
required=True,
type=str,
help="Filename of Yaml file of experiments to run.",
)
def main(experiments_filename: str) -> None:
"""Parse command-line arguments and run experiments from provided file."""
run_experiments(experiments_filename)
if __name__ == "__main__":
main()
|
