This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 17:53:25 +03:00
parent 06878e7656
commit 8f3059be0c
+1 -1
View File
@@ -15,7 +15,7 @@ from security import Security
result_dir = '/azaion/dev/ai-training/runs/detect/train12'
model_dir = path.join(models_dir, f'{prefix}2025-05-18')
shutil.copytree(result_dir, model_dir)
shutil.copytree(result_dir, model_dir, dirs_exist_ok=True)
model_path = path.join(models_dir, f'{prefix[:-1]}.pt')
shutil.copy(path.join(model_dir, 'weights', 'best.pt'), model_path)