vampnet / scripts /utils /stage.py
Hugo Flores
first commit
50f034f
raw
history blame
712 Bytes
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()