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

This commit is contained in:
2026-09-04 09:19:15 +00:00
parent 6f00074199
commit 6b11128357
+5 -2
View File
@@ -11,6 +11,8 @@ _base_:
method: simclr method: simclr
num_clusters: 10
simclr: simclr:
backbone: backbone:
# _replace_ 必须保留:不加的话 base 里 tiny_conv 的 widths/depths/block 会残留, # _replace_ 必须保留:不加的话 base 里 tiny_conv 的 widths/depths/block 会残留,
@@ -31,5 +33,6 @@ simclr:
feature_stages: [5, 6] feature_stages: [5, 6]
# resnet18 比 tiny_conv 重得多(11.2M vs 0.3M),显存和耗时都要留余量 # resnet18 比 tiny_conv 重得多(11.2M vs 0.3M),显存和耗时都要留余量
batch_size: 128 batch_size: 1024
epochs: 40 epochs: 300
optimizer: {name: adamw, lr: 0.01, weight_decay: 0.0001}