diff --git a/configs/supervised_vitb16_448.yaml b/configs/supervised_vitb16_448.yaml
index e0f7cbc..300cbb9 100644
--- a/configs/supervised_vitb16_448.yaml
+++ b/configs/supervised_vitb16_448.yaml
@@ -27,14 +27,14 @@ mask:
 optimization:
   optimizer: adamw
   freeze_weights: true
-  epochs: 300
+  epochs: 1
   lr: 0.01
   weight_decay: 5.0e-4
   lr_schedule: cosine
   gradient_accumulation_steps: 8
   final_lr: 0.0
   early_stopping:
-    enabled: true
+    enabled: false
     patience: 10
     min_delta: 1.0e-4
     min_epochs: 15
diff --git a/configs/supervised_vitg16_224_in22k.yaml b/configs/supervised_vitg16_224_in22k.yaml
index fc58d0f..2257cea 100644
--- a/configs/supervised_vitg16_224_in22k.yaml
+++ b/configs/supervised_vitg16_224_in22k.yaml
@@ -29,14 +29,14 @@ mask:
 optimization:
   optimizer: adamw
   freeze_weights: true
-  epochs: 300
+  epochs: 1
   lr: 0.001
   weight_decay: 5.0e-4
   lr_schedule: cosine
   gradient_accumulation_steps: 16
   final_lr: 0.0
   early_stopping:
-    enabled: true
+    enabled: false
     patience: 10
     min_delta: 1.0e-4
     min_epochs: 15
diff --git a/configs/supervised_vith14_224.yaml b/configs/supervised_vith14_224.yaml
index b0d312f..da8af64 100644
--- a/configs/supervised_vith14_224.yaml
+++ b/configs/supervised_vith14_224.yaml
@@ -27,14 +27,14 @@ mask:
 optimization:
   optimizer: adamw
   freeze_weights: true
-  epochs: 300
+  epochs: 1
   lr: 0.001
   weight_decay: 5.0e-4
   lr_schedule: cosine
   gradient_accumulation_steps: 32
   final_lr: 0.0
   early_stopping:
-    enabled: true
+    enabled: false
     patience: 10
     min_delta: 1.0e-4
     min_epochs: 15
diff --git a/configs/supervised_vith14_224_in1k.yaml b/configs/supervised_vith14_224_in1k.yaml
index b1de2af..30276ac 100644
--- a/configs/supervised_vith14_224_in1k.yaml
+++ b/configs/supervised_vith14_224_in1k.yaml
@@ -28,14 +28,14 @@ mask:
 optimization:
   optimizer: adamw
   freeze_weights: true
-  epochs: 300
+  epochs: 1
   lr: 0.01
   weight_decay: 0
   lr_schedule: cosine
   gradient_accumulation_steps: 64
   final_lr: 0.0
   early_stopping:
-    enabled: true
+    enabled: false
     patience: 10
     min_delta: 1.0e-4
     min_epochs: 15
diff --git a/configs/supervised_vith14_224_in22k.yaml b/configs/supervised_vith14_224_in22k.yaml
index d9a59e7..30ef4c7 100644
--- a/configs/supervised_vith14_224_in22k.yaml
+++ b/configs/supervised_vith14_224_in22k.yaml
@@ -28,14 +28,14 @@ mask:
 optimization:
   optimizer: adamw
   freeze_weights: true
-  epochs: 300
+  epochs: 1
   lr: 0.001
   weight_decay: 5.0e-4
   lr_schedule: cosine
   gradient_accumulation_steps: 32
   final_lr: 0.0
   early_stopping:
-    enabled: true
+    enabled: false
     patience: 10
     min_delta: 1.0e-4
     min_epochs: 15
diff --git a/configs/supervised_vith16_448.yaml b/configs/supervised_vith16_448.yaml
index 3e4b95f..0f3a702 100644
--- a/configs/supervised_vith16_448.yaml
+++ b/configs/supervised_vith16_448.yaml
@@ -27,14 +27,14 @@ mask:
 optimization:
   optimizer: adamw
   freeze_weights: true
-  epochs: 300
+  epochs: 1
   lr: 0.01
   weight_decay: 5.0e-4
   lr_schedule: cosine
   gradient_accumulation_steps: 64
   final_lr: 0.0
   early_stopping:
-    enabled: true
+    enabled: false
     patience: 10
     min_delta: 1.0e-4
     min_epochs: 15
diff --git a/configs/supervised_vith16_448_in1k.yaml b/configs/supervised_vith16_448_in1k.yaml
index 5d0bd85..4a219f5 100644
--- a/configs/supervised_vith16_448_in1k.yaml
+++ b/configs/supervised_vith16_448_in1k.yaml
@@ -29,14 +29,14 @@ mask:
 optimization:
   optimizer: adamw
   freeze_weights: true
-  epochs: 300
+  epochs: 1
   lr: 0.001
   weight_decay: 0
   lr_schedule: cosine
   gradient_accumulation_steps: 32
   final_lr: 0.0
   early_stopping:
-    enabled: true
+    enabled: false
     patience: 10
     min_delta: 1.0e-4
     min_epochs: 15