fix(configs): update num_clusters and training parameters in film_simclr_resnet18.yaml

2026-09-04 13:41:41 +00:00
parent 865cdd7210
commit 703ba7076b
2 changed files with 1334 additions and 1334 deletions
@@ -11,7 +11,7 @@ _base_:

 method: simclr

-num_clusters: 10
+num_clusters: 5

 simclr:
  backbone:
@@ -34,5 +34,5 @@ simclr:

  # resnet18 比 tiny_conv 重得多（11.2M vs 0.3M），显存和耗时都要留余量
  batch_size: 1024
-  epochs: 300
-  optimizer: {name: adamw, lr: 0.01, weight_decay: 0.0001}
+  epochs: 1700
+  optimizer: {name: adamw, lr: 0.001, weight_decay: 0.0001}