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

uncommenting

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 18:12:52 +03:00
parent 41d62ffc35
commit 0bd5bd6d27
2 changed files with 6 additions and 6 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') # resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt')
# result_dir = '/azaion/dev/ai-training/runs/detect/train12' result_dir = '/azaion/dev/ai-training/runs/detect/train12'
# model_dir = path.join(models_dir, f'{prefix}2025-05-18') model_dir = path.join(models_dir, f'{prefix}2025-05-18')
# shutil.copytree(result_dir, model_dir, dirs_exist_ok=True) shutil.copytree(result_dir, model_dir, dirs_exist_ok=True)
#
model_path = path.join(models_dir, f'{prefix[:-1]}.pt') 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() api_client = ApiClient()
onnx_path = export_onnx(model_path) onnx_path = export_onnx(model_path)
requirements.txt
+1 -1
View File
@@ -19,4 +19,4 @@ onnxruntime-gpu
netron netron
pycuda pycuda
tensorrt tensorrt
onnx onnx