fix(configs): update film_simclr_resnet18.yaml with new clustering parameters and optimizer settings

2026-09-04 09:19:15 +00:00
parent 6f00074199
commit 6b11128357
1 changed files with 5 additions and 2 deletions
@@ -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}