Added validation for epochs and batchsize

backend/aiproject/settings.py

@@ -53,6 +53,11 @@
 OSM_SECRET_KEY = env("OSM_SECRET_KEY")
 
 
+# Limiter
+EPOCHS_LIMIT = env("EPOCHS_LIMIT", default=30)
+BATCH_SIZE_LIMIT = env("BATCH_SIZE_LIMIT", default=8)
+
+
 # Application definition
 
 INSTALLED_APPS = [

backend/core/views.py

@@ -118,6 +118,18 @@ def create(self, validated_data):
                 "Another training is already running or submitted for this model."
             )
 
+        epochs = validated_data["epochs"]
+        batch_size = validated_data["batch_size"]
+
+        if epochs > settings.EPOCHS_LIMIT:
+            raise ValidationError(
+                f"Epochs can't be greater than {settings.EPOCHS_LIMIT} on this server"
+            )
+        if batch_size > settings.BATCH_SIZE_LIMIT:
+            raise ValidationError(
+                f"Batch size can't be greater than {settings.BATCH_SIZE_LIMIT} on this server"
+            )
+
         user = self.context["request"].user
         validated_data["created_by"] = user
         # create the model instance