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

add switch to save each epoch

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2024-06-19 23:16:45 +03:00
parent eba8b62db8
commit 3fd72202f2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -1
View File
@@ -106,7 +106,7 @@ if __name__ == '__main__':
folder = f'{prefix}2024-06-18' folder = f'{prefix}2024-06-18'
yaml = abspath(path.join('datasets', folder, 'data.yaml')) yaml = abspath(path.join('datasets', folder, 'data.yaml'))
results = model.train(data=yaml, epochs=100, batch=10, imgsz=640) results = model.train(data=yaml, epochs=100, batch=10, imgsz=640, save_period=1)
shutil.copy(f'{results.save_dir}/weights/best.pt', latest_model) shutil.copy(f'{results.save_dir}/weights/best.pt', latest_model)
shutil.copytree(results.save_dir, f'models/{folder}') shutil.copytree(results.save_dir, f'models/{folder}')