From 56cdf38d8c4460438987110654b489e64819da60 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Thu, 21 May 2026 16:05:26 +0200 Subject: [PATCH] try new params --- configs/eval_wilds_vith14.yaml | 6 +++--- configs/supervised_wilds_vith14_ep300-lp.yaml | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/configs/eval_wilds_vith14.yaml b/configs/eval_wilds_vith14.yaml index 84c7f36..a20f513 100644 --- a/configs/eval_wilds_vith14.yaml +++ b/configs/eval_wilds_vith14.yaml @@ -6,8 +6,8 @@ meta: patch_size: 14 crop_size: 224 use_bfloat16: true - checkpoint_folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.001-wd0.0005/ - read_checkpoint: linear_probe-lars-lr0.001-wd0.0005-ep50.pth.tar + checkpoint_folder: experiment_logs/supervised/ + read_checkpoint: linear_probe.pth.tar data: batch_size: 128 @@ -18,5 +18,5 @@ data: download: true logging: - folder: experiment_logs/eval-wilds/vith14.224-bs.128-ep.300-ft/ + folder: experiment_logs/eval-wilds/vith14-ep300-adamw-ep.50-lr0.01-wd0.0005/ write_tag: iwildcam_test diff --git a/configs/supervised_wilds_vith14_ep300-lp.yaml b/configs/supervised_wilds_vith14_ep300-lp.yaml index 81380e7..88dee12 100644 --- a/configs/supervised_wilds_vith14_ep300-lp.yaml +++ b/configs/supervised_wilds_vith14_ep300-lp.yaml @@ -24,7 +24,7 @@ mask: patch_size: 14 optimization: - optimizer: lars # 'adamw', 'sgd', or 'lars' + optimizer: adamw # 'adamw', 'sgd', or 'lars' freeze_weights: true # true for linear probing, false for full fine-tuning epochs: 50 lr: 0.01 @@ -55,5 +55,5 @@ validation: eval_every: 1 logging: - folder: experiment_logs/supervised-vith14.224-bs.256-ep.50-lp-lars-lr0.01-wd0.0005/ - write_tag: linear_probe-lars-lr0.01-wd0.0005 + folder: experiment_logs/supervised/ + write_tag: linear_probe