fix(configs): update optimizer settings and add projection parameters in film_simclr_resnet50.yaml

2026-09-07 06:25:54 +00:00
parent 0f5bb4a5c0
commit 01cfff9b02
1 changed files with 2 additions and 1 deletions
@@ -39,7 +39,8 @@ simclr:
  batch_size: 256

  epochs: 2500
-  optimizer: {name: adamw, lr: 0.0001, weight_decay: 0.00001}
+  projection: {hidden_channels: 2048, out_channels: 128}
+  optimizer: {name: adamw, lr: 0.001, weight_decay: 0.0001}

  # 投影头沿用 base 的 256 隐藏维。SimCLR 原文给 resnet50 用的是 2048->2048->128，
  # 想更贴近原设置可以改成 {hidden_channels: 2048, out_channels: 128}——