diff --git a/configs/cluster/film_simclr_resnet18.yaml b/configs/cluster/film_simclr_resnet18.yaml index a699219..3cc318c 100644 --- a/configs/cluster/film_simclr_resnet18.yaml +++ b/configs/cluster/film_simclr_resnet18.yaml @@ -34,5 +34,5 @@ simclr: # resnet18 比 tiny_conv 重得多(11.2M vs 0.3M),显存和耗时都要留余量 batch_size: 1024 - epochs: 1700 - optimizer: {name: adamw, lr: 0.001, weight_decay: 0.0001} + epochs: 2500 + optimizer: {name: adamw, lr: 0.0001, weight_decay: 0.00001}