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

fix manual

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 19:04:16 +03:00
parent 44c9e87bd4
commit c9f2c167e2
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
manual_run.py
-2
View File
@@ -11,10 +11,8 @@ from augmentation import Augmentator
# train.train_dataset() # train.train_dataset()
# train.resume_training('/azaion/dev/ai-training/runs/detect/train12/weights/last.pt') # train.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(constants.models_dir, f'{constants.prefix}2025-05-18') model_dir = path.join(constants.models_dir, f'{constants.prefix}2025-05-18')
shutil.copytree(result_dir, model_dir, dirs_exist_ok=True)
for file in glob.glob(path.join(model_dir, 'weights', 'epoch*')): for file in glob.glob(path.join(model_dir, 'weights', 'epoch*')):
os.remove(file) os.remove(file)
shutil.copy(path.join(model_dir, 'weights', 'best.pt'), constants.CURRENT_PT_MODEL) shutil.copy(path.join(model_dir, 'weights', 'best.pt'), constants.CURRENT_PT_MODEL)