ad39d509e04a5aa00462e441ec31ca1b24f15491,python/baseline/pytorch/classify/train.py,ClassifyTrainerPyTorch,__init__,#ClassifyTrainerPyTorch#Any#,19
Before Change
else:
mom = kwargs.get("mom", 0.9)
print("using mom [%.3f]" % mom)
self.optimizer = torch.optim.SGD(parameters, lr=eta, momentum=mom)
self.crit = model.create_loss().cuda()
self.model = torch.nn.DataParallel(model).cuda()
After Change
else:
mom = kwargs.get("mom", 0.9)
print("using mom [%.3f]" % mom)
self.optimizer = torch.optim.SGD(parameters, lr=eta, momentum=mom, weight_decay=weight_decay)
self.crit = model.create_loss().cuda()
self.model = torch.nn.DataParallel(model).cuda()
In pattern: SUPERPATTERN
Frequency: 3
Non-data size: 4
Instances
Project Name: dpressel/mead-baseline
Commit Name: ad39d509e04a5aa00462e441ec31ca1b24f15491
Time: 2018-06-29
Author: dpressel@gmail.com
File Name: python/baseline/pytorch/classify/train.py
Class Name: ClassifyTrainerPyTorch
Method Name: __init__
Project Name: horovod/horovod
Commit Name: 5fa1d7aea2c89fdfa15a688aa413ea49a480ab38
Time: 2019-11-25
Author: tix@microsoft.com
File Name: examples/pytorch_imagenet_resnet50.py
Class Name:
Method Name:
Project Name: deepgram/kur
Commit Name: 45406badf091b037f871a1efae3fbdd663b9cef7
Time: 2017-01-18
Author: ajsyp@syptech.net
File Name: kur/optimizer/sgd.py
Class Name: SGD
Method Name: get_optimizer