diff --git a/configs/cluster/film_simclr_resnet50.yaml b/configs/cluster/film_simclr_resnet50.yaml index 2c274c3..a46eb9c 100644 --- a/configs/cluster/film_simclr_resnet50.yaml +++ b/configs/cluster/film_simclr_resnet50.yaml @@ -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}——