simplify module3 venue workflow

This commit is contained in:
2026-09-16 06:33:11 +03:00
parent 25972390f1
commit 68dde2cfaf
6 changed files with 58 additions and 33 deletions
+2 -2
View File
@@ -31,7 +31,7 @@ def select_device(requested, torch):
def arguments():
parser = argparse.ArgumentParser()
parser.add_argument("--data", default="training/data.yaml", help="data.yaml из Roboflow")
parser.add_argument("--data", default="training/data.yaml", help="YOLO data.yaml")
parser.add_argument("--model", default="yolo11n.pt", help="Стартовые веса Ultralytics")
parser.add_argument("--epochs", type=int, default=30)
parser.add_argument(
@@ -62,7 +62,7 @@ def main():
data = local_path(args.data)
if not data.is_file():
raise SystemExit(f"Не найден датасет: {data}\nРаспакуйте YOLO11 export из Roboflow в module3/dataset.")
raise SystemExit(f"Не найден датасет: {data}\nПодготовьте YOLO-датасет по module3/SIMPLE.md.")
device = select_device(args.device, torch)
if device == 0:
backend = "ROCm" if getattr(torch.version, "hip", None) else "CUDA"