mumble-cleanup / scripts /06_pack_and_ship.py
adikuma's picture
initial upload: cleanup code and 688-pair seed dataset
fd0b01f verified
Raw
History Blame Contribute Delete
694 Bytes
# tar runs/<run-id>/ with model + onnx + eval json + manifest, print sha256
# and the vastai copy command for transfer back to the laptop.
import argparse
from pathlib import Path
from cleanup.pack.ship import pack
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("--runs-dir", default="runs")
parser.add_argument("--dist-dir", default="dist")
parser.add_argument("--run-id", required=True)
args = parser.parse_args()
run_dir = Path(args.runs_dir) / args.run_id
if not run_dir.exists():
raise FileNotFoundError(f"no run at {run_dir}")
pack(run_dir, Path(args.dist_dir), args.run_id)
if __name__ == "__main__":
main()