diff --git a/configs/cluster/film_simclr_resnet18.yaml b/configs/cluster/film_simclr_resnet18.yaml index 21a049f..e2e3e1d 100644 --- a/configs/cluster/film_simclr_resnet18.yaml +++ b/configs/cluster/film_simclr_resnet18.yaml @@ -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}