Change the defaults based on the model base

hotosm · Jan 20, 2025 · 8400295 · 8400295
1 parent cf688da
commit 8400295
Showing 1 changed file with 3 additions and 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
@@ -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,
     },