From 8400295edbb6c5cf347c8384ba8f2812d5717bd6 Mon Sep 17 00:00:00 2001
From: Omran NAJJAR <omran.najjar@gmail.com>
Date: Mon, 20 Jan 2025 16:18:43 +0100
Subject: [PATCH] Change the defaults based on the model base

---
 .../components/training-settings/training-settings-form.tsx | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/frontend/src/features/model-creation/components/training-settings/training-settings-form.tsx b/frontend/src/features/model-creation/components/training-settings/training-settings-form.tsx
index aa4b5af6..759784e7 100644
--- a/frontend/src/features/model-creation/components/training-settings/training-settings-form.tsx
+++ b/frontend/src/features/model-creation/components/training-settings/training-settings-form.tsx
@@ -64,19 +64,19 @@ const TrainingSettingsForm = () => {
   const defaultTrainingSettings = {
     [TrainingType.BASIC]: {
       epoch: formData.baseModel === BASE_MODELS.RAMP ? 2 : 20,
-      batchSize: formData.baseModel === BASE_MODELS.RAMP ? 4 : 8,
+      batchSize: formData.baseModel === BASE_MODELS.RAMP ? 4 : 4,
       contactSpacing: 8,
       boundaryWidth: 3,
     },
     [TrainingType.INTERMEDIATE]: {
       epoch: formData.baseModel === BASE_MODELS.RAMP ? 20 : 50,
-      batchSize: formData.baseModel === BASE_MODELS.RAMP ? 8 : 12,
+      batchSize: formData.baseModel === BASE_MODELS.RAMP ? 8 : 8,
       contactSpacing: 8,
       boundaryWidth: 3,
     },
     [TrainingType.ADVANCED]: {
       epoch: formData.baseModel === BASE_MODELS.RAMP ? 30 : 150,
-      batchSize: formData.baseModel === BASE_MODELS.RAMP ? 12 : 16,
+      batchSize: formData.baseModel === BASE_MODELS.RAMP ? 8 : 8,
       contactSpacing: 8,
       boundaryWidth: 3,
     },