azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 23:26:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fixes, hash fix

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 18:06:14 +03:00
parent 8f3059be0c
commit c0a671a60b
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
manual_run.py
+5 -5
View File
@@ -13,12 +13,12 @@ from security import Security
# resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')
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, dirs_exist_ok=True)
# 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, 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)
# shutil.copy(path.join(model_dir, 'weights', 'best.pt'), model_path)
api_client = ApiClient()
onnx_path = export_onnx(model_path)