(net)
| 186 | |
| 187 | |
| 188 | def wd_loss(net): |
| 189 | loss = 0 |
| 190 | for name, param in net.named_parameters(): |
| 191 | if ('bn' in name or 'bias' in name): |
| 192 | continue |
| 193 | elif ('weight' in name): |
| 194 | loss = loss + torch.sum(param ** 2) / 2 |
| 195 | return loss |
| 196 | |
| 197 | |
| 198 | def get_optimizer(net, optim_name='SGD', lr=0.1, momentum=0.9, weight_decay=0, nesterov=True, bn_wd_skip=True): |
nothing calls this directly
no outgoing calls
no test coverage detected