iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update configs

Browse Source
This commit is contained in:
Xuanyi Dong
2019-03-31 23:00:37 +08:00
parent 4bac459bf9
commit 322d5462a3
5 changed files with 31 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps-cnn/train_utils.py
View File

@@ -54,7 +54,7 @@ def main_procedure(config, dataset, data_path, args, genotype, init_channels, la
optimizer = torch.optim.SGD(model.parameters(), config.LR, momentum=config.momentum, weight_decay=config.decay)
#optimizer = torch.optim.SGD(model.parameters(), config.LR, momentum=config.momentum, weight_decay=config.decay, nestero=True)
if config.type == 'cosine':
scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, float(config.epochs))
scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(optimizer, float(config.epochs), eta_min=float(config.LR_MIN))
else:
raise ValueError('Can not find the schedular type : {:}'.format(config.type))