YannAhlgrim/wilds-ijepa
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

change epoch to best epoch + disable early stopping

Browse Source
This commit is contained in:
YannAhlgrim
2026-07-19 17:59:49 +02:00
parent 0cc2f3532b
commit 04b825af2a
7 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/supervised_vitb16_448.yaml
+2 -2
View File
@@ -27,14 +27,14 @@ mask:
optimization: optimization:
optimizer: adamw optimizer: adamw
freeze_weights: true freeze_weights: true
epochs: 300 epochs: 1
lr: 0.01 lr: 0.01
weight_decay: 5.0e-4 weight_decay: 5.0e-4
lr_schedule: cosine lr_schedule: cosine
gradient_accumulation_steps: 8 gradient_accumulation_steps: 8
final_lr: 0.0 final_lr: 0.0
early_stopping: early_stopping:
enabled: true enabled: false
patience: 10 patience: 10
min_delta: 1.0e-4 min_delta: 1.0e-4
min_epochs: 15 min_epochs: 15
configs/supervised_vitg16_224_in22k.yaml
+2 -2
View File
@@ -29,14 +29,14 @@ mask:
optimization: optimization:
optimizer: adamw optimizer: adamw
freeze_weights: true freeze_weights: true
epochs: 300 epochs: 1
lr: 0.001 lr: 0.001
weight_decay: 5.0e-4 weight_decay: 5.0e-4
lr_schedule: cosine lr_schedule: cosine
gradient_accumulation_steps: 16 gradient_accumulation_steps: 16
final_lr: 0.0 final_lr: 0.0
early_stopping: early_stopping:
enabled: true enabled: false
patience: 10 patience: 10
min_delta: 1.0e-4 min_delta: 1.0e-4
min_epochs: 15 min_epochs: 15
configs/supervised_vith14_224.yaml
+2 -2
View File
@@ -27,14 +27,14 @@ mask:
optimization: optimization:
optimizer: adamw optimizer: adamw
freeze_weights: true freeze_weights: true
epochs: 300 epochs: 1
lr: 0.001 lr: 0.001
weight_decay: 5.0e-4 weight_decay: 5.0e-4
lr_schedule: cosine lr_schedule: cosine
gradient_accumulation_steps: 32 gradient_accumulation_steps: 32
final_lr: 0.0 final_lr: 0.0
early_stopping: early_stopping:
enabled: true enabled: false
patience: 10 patience: 10
min_delta: 1.0e-4 min_delta: 1.0e-4
min_epochs: 15 min_epochs: 15
configs/supervised_vith14_224_in1k.yaml
+2 -2
View File
@@ -28,14 +28,14 @@ mask:
optimization: optimization:
optimizer: adamw optimizer: adamw
freeze_weights: true freeze_weights: true
epochs: 300 epochs: 1
lr: 0.01 lr: 0.01
weight_decay: 0 weight_decay: 0
lr_schedule: cosine lr_schedule: cosine
gradient_accumulation_steps: 64 gradient_accumulation_steps: 64
final_lr: 0.0 final_lr: 0.0
early_stopping: early_stopping:
enabled: true enabled: false
patience: 10 patience: 10
min_delta: 1.0e-4 min_delta: 1.0e-4
min_epochs: 15 min_epochs: 15
configs/supervised_vith14_224_in22k.yaml
+2 -2
View File
@@ -28,14 +28,14 @@ mask:
optimization: optimization:
optimizer: adamw optimizer: adamw
freeze_weights: true freeze_weights: true
epochs: 300 epochs: 1
lr: 0.001 lr: 0.001
weight_decay: 5.0e-4 weight_decay: 5.0e-4
lr_schedule: cosine lr_schedule: cosine
gradient_accumulation_steps: 32 gradient_accumulation_steps: 32
final_lr: 0.0 final_lr: 0.0
early_stopping: early_stopping:
enabled: true enabled: false
patience: 10 patience: 10
min_delta: 1.0e-4 min_delta: 1.0e-4
min_epochs: 15 min_epochs: 15
configs/supervised_vith16_448.yaml
+2 -2
View File
@@ -27,14 +27,14 @@ mask:
optimization: optimization:
optimizer: adamw optimizer: adamw
freeze_weights: true freeze_weights: true
epochs: 300 epochs: 1
lr: 0.01 lr: 0.01
weight_decay: 5.0e-4 weight_decay: 5.0e-4
lr_schedule: cosine lr_schedule: cosine
gradient_accumulation_steps: 64 gradient_accumulation_steps: 64
final_lr: 0.0 final_lr: 0.0
early_stopping: early_stopping:
enabled: true enabled: false
patience: 10 patience: 10
min_delta: 1.0e-4 min_delta: 1.0e-4
min_epochs: 15 min_epochs: 15
configs/supervised_vith16_448_in1k.yaml
+2 -2
View File
@@ -29,14 +29,14 @@ mask:
optimization: optimization:
optimizer: adamw optimizer: adamw
freeze_weights: true freeze_weights: true
epochs: 300 epochs: 1
lr: 0.001 lr: 0.001
weight_decay: 0 weight_decay: 0
lr_schedule: cosine lr_schedule: cosine
gradient_accumulation_steps: 32 gradient_accumulation_steps: 32
final_lr: 0.0 final_lr: 0.0
early_stopping: early_stopping:
enabled: true enabled: false
patience: 10 patience: 10
min_delta: 1.0e-4 min_delta: 1.0e-4
min_epochs: 15 min_epochs: 15