diff --git a/configs/supervised_wilds_vith14_ep300-lp.yaml b/configs/supervised_wilds_vith14_ep300-lp.yaml index b1ffe61..23aa965 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: sgd # 'adamw', 'sgd', or 'lars' + optimizer: lars # 'adamw', 'sgd', or 'lars' freeze_weights: true # true for linear probing, false for full fine-tuning epochs: 300 lr: 0.01