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}')