(self, params, lr=required, momentum=0, dampening=0,
weight_decay=0, nesterov=False)
| 68 | """ |
| 69 | |
| 70 | def __init__(self, params, lr=required, momentum=0, dampening=0, |
| 71 | weight_decay=0, nesterov=False): |
| 72 | if lr is not required and lr < 0.0: |
| 73 | raise ValueError("Invalid learning rate: {}".format(lr)) |
| 74 | if momentum < 0.0: |
| 75 | raise ValueError("Invalid momentum value: {}".format(momentum)) |
| 76 | if weight_decay < 0.0: |
| 77 | raise ValueError("Invalid weight_decay value: {}".format(weight_decay)) |
| 78 | |
| 79 | defaults = dict(lr=lr, momentum=momentum, dampening=dampening, |
| 80 | weight_decay=weight_decay, nesterov=nesterov) |
| 81 | if nesterov and (momentum <= 0 or dampening != 0): |
| 82 | raise ValueError("Nesterov momentum requires a momentum and zero dampening") |
| 83 | super(SGD, self).__init__(params, defaults) |
| 84 | |
| 85 | def __setstate__(self, state): |
| 86 | super(SGD, self).__setstate__(state) |
nothing calls this directly
no outgoing calls
no test coverage detected