From 13dee5b8cd0e8bacd88cacc2319838e702d88f7e Mon Sep 17 00:00:00 2001 From: Aherobo Ovie Victor Date: Tue, 22 Jul 2025 12:28:57 +0100 Subject: [PATCH] code reviewed --- train.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/train.py b/train.py index 794cadd..4830601 100644 --- a/train.py +++ b/train.py @@ -36,8 +36,13 @@ def train_model(): cache='disk', # Changed to disk caching for deterministic results ) + # Create model directory if it doesn't exist + import os + os.makedirs('model/weights', exist_ok=True) + # Save the trained model model.save('model/weights/best.pt') + if __name__ == '__main__': train_model() \ No newline at end of file