From 6b1112835714252c9cf95cbf25fa1dbd4ccb649c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=91=A8=E7=BF=94=E5=AE=87?= Date: Fri, 4 Sep 2026 09:19:15 +0000 Subject: [PATCH] fix(configs): update film_simclr_resnet18.yaml with new clustering parameters and optimizer settings --- configs/cluster/film_simclr_resnet18.yaml | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/configs/cluster/film_simclr_resnet18.yaml b/configs/cluster/film_simclr_resnet18.yaml index 21a049f..e2e3e1d 100644 --- a/configs/cluster/film_simclr_resnet18.yaml +++ b/configs/cluster/film_simclr_resnet18.yaml @@ -11,6 +11,8 @@ _base_: method: simclr +num_clusters: 10 + simclr: backbone: # _replace_ 必须保留：不加的话 base 里 tiny_conv 的 widths/depths/block 会残留， @@ -31,5 +33,6 @@ simclr: feature_stages: [5, 6] # resnet18 比 tiny_conv 重得多（11.2M vs 0.3M），显存和耗时都要留余量 - batch_size: 128 - epochs: 40 + batch_size: 1024 + epochs: 300 + optimizer: {name: adamw, lr: 0.01, weight_decay: 0.0001}