gloamxun/benchmarkexp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fix the mistake of the parameters

Browse Source
This commit is contained in:
YusenZhao
2026-02-06 15:15:32 +08:00
parent 06cc14c47a
commit ecad6bcd4a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dl-main.py
View File

@@ -90,9 +90,9 @@ def run_cifar10(model_name='ResNet-34'):
dmadam_eta0 = 0.9 if model_name == 'VGG-13' else 1.4 dmadam_eta0 = 0.9 if model_name == 'VGG-13' else 1.4
optimizers_config = { optimizers_config = {
'SGD': {'lr': 0.1, 'weight_decay': 0.0005}, 'SGD': {'lr': 0.001, 'weight_decay': 0.0005},
'SGDM': {'lr': 0.1, 'momentum': 0.9, 'weight_decay': 0.0005}, 'SGDM': {'lr': 0.001, 'momentum': 0.9, 'weight_decay': 0.0005},
'Adagrad': {'lr': 0.01, 'weight_decay': 0.0005}, 'Adagrad': {'lr': 0.001, 'weight_decay': 0.0005},
'Adam': {'lr': 0.001, 'betas': (0.9, 0.999), 'weight_decay': 0.0005}, 'Adam': {'lr': 0.001, 'betas': (0.9, 0.999), 'weight_decay': 0.0005},
'AdamW': {'lr': 0.001, 'betas': (0.9, 0.999), 'weight_decay': 0.0005}, 'AdamW': {'lr': 0.001, 'betas': (0.9, 0.999), 'weight_decay': 0.0005},
'DMAdam': {'lr': 0.001, 'eta0': dmadam_eta0, 'eps': 1e-8} 'DMAdam': {'lr': 0.001, 'eta0': dmadam_eta0, 'eps': 1e-8}