Buckets:
| """Thin OPD command-line entrypoint.""" | |
| from __future__ import annotations | |
| import argparse | |
| from .config import load_config | |
| def main() -> None: | |
| parser = argparse.ArgumentParser(description=__doc__) | |
| parser.add_argument("command", choices=("train", "inspect", "audit-rgb")) | |
| parser.add_argument("--config", required=True) | |
| parser.add_argument("--set", action="append", default=[]) | |
| parser.add_argument("--output-dir") | |
| parser.add_argument("--samples", type=int, default=1) | |
| parser.add_argument("--decoder-steps", type=int, default=28) | |
| parser.add_argument("--stop-after-updates", type=int, help="commit and pause after N new updates without changing the training horizon") | |
| args = parser.parse_args() | |
| config = load_config(args.config, args.set) | |
| if args.command == "inspect": | |
| import json | |
| print(json.dumps(config.as_dict(), indent=2)) | |
| return | |
| if args.command == "audit-rgb": | |
| from .audit import run_rgb_audit | |
| run_rgb_audit( | |
| config, | |
| output_dir=args.output_dir, | |
| samples=args.samples, | |
| decoder_steps=args.decoder_steps, | |
| ) | |
| return | |
| if args.output_dir is not None: | |
| raise ValueError("--output-dir is only valid for audit-rgb") | |
| from .trainer import train | |
| train(config, stop_after_updates=args.stop_after_updates) | |
| __all__ = ["main"] | |
Xet Storage Details
- Size:
- 1.39 kB
- Xet hash:
- d721ad945bc834d44123d47b1a5d150e196b011ed916f5a72a6f670c3c9d2a76
·
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.