fix(configs): update film_simclr_resnet18.yaml with new clustering parameters and optimizer settings
This commit is contained in:
@@ -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}
|
||||||
|
|||||||
Reference in New Issue
Block a user