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)