From 3fd72202f2a86d278570842c54277c260bd181a7 Mon Sep 17 00:00:00 2001 From: Alex Bezdieniezhnykh Date: Wed, 19 Jun 2024 23:16:45 +0300 Subject: [PATCH] add switch to save each epoch --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index ae6ef10..dbe12e7 100644 --- a/train.py +++ b/train.py @@ -106,7 +106,7 @@ if __name__ == '__main__': folder = f'{prefix}2024-06-18' 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.copytree(results.save_dir, f'models/{folder}')