Spaces:
Sleeping
Sleeping
File size: 712 Bytes
50f034f |
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 |
import os
import subprocess
from pathlib import Path
import argbind
import rich
from audiotools.ml import Experiment
@argbind.bind(without_prefix=True)
def run(
run_dir: str = os.getenv("PATH_TO_RUNS", "runs"),
name: str = None,
recent: bool = False,
):
if recent:
paths = sorted(Path(run_dir).iterdir(), key=os.path.getmtime)
paths = [p.name for p in paths if p.is_dir()]
if paths:
name = paths[-1]
with Experiment(run_dir, name) as exp:
exp.snapshot()
rich.print(f"Created a snapshot of {exp.parent_directory} at {exp.exp_dir}")
if __name__ == "__main__":
args = argbind.parse_args()
with argbind.scope(args):
run()
|