From 04b825af2a81d242fa3df7ec63ec62084c855276 Mon Sep 17 00:00:00 2001
From: YannAhlgrim <yann@ahlgrim.net>
Date: Sun, 19 Jul 2026 17:59:49 +0200
Subject: [PATCH] change epoch to best epoch + disable early stopping

---
 configs/supervised_vitb16_448.yaml       | 4 ++--
 configs/supervised_vitg16_224_in22k.yaml | 4 ++--
 configs/supervised_vith14_224.yaml       | 4 ++--
 configs/supervised_vith14_224_in1k.yaml  | 4 ++--
 configs/supervised_vith14_224_in22k.yaml | 4 ++--
 configs/supervised_vith16_448.yaml       | 4 ++--
 configs/supervised_vith16_448_in1k.yaml  | 4 ++--
 7 files changed, 14 insertions(+), 14 deletions(-)

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