From 02be49c4f17c1b06baf191887c546eda58e116a0 Mon Sep 17 00:00:00 2001 From: kshitijrajsharma Date: Fri, 6 Dec 2024 13:19:27 +0100 Subject: [PATCH] fix(settings): add global variables for epochs and batch size limits --- backend/aiproject/settings.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/backend/aiproject/settings.py b/backend/aiproject/settings.py index 4ecea1b9..c77b639f 100644 --- a/backend/aiproject/settings.py +++ b/backend/aiproject/settings.py @@ -57,6 +57,10 @@ # Limiter + +EPOCHS_LIMIT = env("EPOCHS_LIMIT", default=20) ## TODO : Remove this global variable +BATCH_SIZE_LIMIT = env("BATCH_SIZE_LIMIT", default=8) + ## YOLO YOLO_EPOCHS_LIMIT = env("YOLO_EPOCHS_LIMIT", default=200) YOLO_BATCH_SIZE_LIMIT = env("YOLO_BATCH_SIZE_LIMIT", default=8)