From 1ceaf09a1365a8ffe9cda763a127ab2f4fa69007 Mon Sep 17 00:00:00 2001 From: YannAhlgrim Date: Mon, 15 Jun 2026 13:52:07 +0200 Subject: [PATCH] grid for vit b ep 600 --- configs/grids/lp_grid.yaml | 6 +++--- configs/supervised_wilds_vitb16-448_ep600-lp.yaml | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/configs/grids/lp_grid.yaml b/configs/grids/lp_grid.yaml index 0685134..66c73d8 100644 --- a/configs/grids/lp_grid.yaml +++ b/configs/grids/lp_grid.yaml @@ -8,8 +8,8 @@ grid: optimization.lr: [0.01, 0.05, 0.001] optimization.weight_decay: [5.0e-4, 0] optimization.momentum: [0.9, 0.5] - optimization.lr_schedule: [cosine, step] - data.batch_size: [64, 128, 256] + optimization.lr_schedule: [cosine] + data.batch_size: [256, 512, 1024] meta.representation_type: [last_avgpool, last4_avgpool_concat] meta.head_type: [linear, bn_linear] @@ -18,4 +18,4 @@ launch: partition: gpu1 nodes: 1 tasks_per_node: 1 - time: 180 + time: 600 diff --git a/configs/supervised_wilds_vitb16-448_ep600-lp.yaml b/configs/supervised_wilds_vitb16-448_ep600-lp.yaml index 5deb634..43971f1 100644 --- a/configs/supervised_wilds_vitb16-448_ep600-lp.yaml +++ b/configs/supervised_wilds_vitb16-448_ep600-lp.yaml @@ -3,7 +3,7 @@ meta: embed_dim: 768 load_checkpoint: true checkpoint_folder: experiment_logs/vitb16.448-bs.16-ep.600/ - read_checkpoint: jepa-ep150.pth.tar + read_checkpoint: jepa-ep600.pth.tar use_bfloat16: true num_classes: 182 representation_type: last_avgpool