simplify module3 venue workflow
This commit is contained in:
+2
-2
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user