lars
This commit is contained in:
@@ -24,7 +24,7 @@ mask:
|
|||||||
patch_size: 14
|
patch_size: 14
|
||||||
|
|
||||||
optimization:
|
optimization:
|
||||||
optimizer: adamw # 'adamw', 'sgd', or 'lars'
|
optimizer: lars # 'adamw', 'sgd', or 'lars'
|
||||||
freeze_weights: true # true for linear probing, false for full fine-tuning
|
freeze_weights: true # true for linear probing, false for full fine-tuning
|
||||||
epochs: 300
|
epochs: 300
|
||||||
lr: 0.01
|
lr: 0.01
|
||||||
|
|||||||
Reference in New Issue
Block a user