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

remove unnecessary middle epochs

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-05-31 19:02:57 +03:00
parent 80c2433141
commit 44c9e87bd4
2 changed files with 19 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
manual_run.py
+8 -5
View File
@@ -1,8 +1,10 @@
import glob
import os
import shutil
from os import path
import constants
import train
from constants import models_dir, prefix
from augmentation import Augmentator
# Augmentator().augment_annotations()
@@ -10,11 +12,12 @@ from augmentation import Augmentator
# 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(models_dir, f'{prefix}2025-05-18')
shutil.copytree(result_dir, model_dir, dirs_exist_ok=True)
model_dir = path.join(constants.models_dir, f'{constants.prefix}2025-05-18')
model_path = path.join(models_dir, f'{prefix[:-1]}.pt')
shutil.copy(path.join(model_dir, 'weights', 'best.pt'), model_path)
shutil.copytree(result_dir, model_dir, dirs_exist_ok=True)
for file in glob.glob(path.join(model_dir, 'weights', 'epoch*')):
os.remove(file)
shutil.copy(path.join(model_dir, 'weights', 'best.pt'), constants.CURRENT_PT_MODEL)
train.export_current_model()
print('success!')