fix(settings): add global variables for epochs and batch size limits

hotosm · Dec 6, 2024 · 02be49c · 02be49c
1 parent eb0eb4f
commit 02be49c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)