(self)
| 416 | return xi, xt |
| 417 | |
| 418 | def param_groups(self): |
| 419 | groups = [{ |
| 420 | 'params': [ |
| 421 | p for n, p in self.named_parameters() |
| 422 | if 'norm' in n or n.endswith('bias') |
| 423 | ], |
| 424 | 'weight_decay': 0.0 |
| 425 | }, { |
| 426 | 'params': [ |
| 427 | p for n, p in self.named_parameters() |
| 428 | if not ('norm' in n or n.endswith('bias')) |
| 429 | ] |
| 430 | }] |
| 431 | return groups |
| 432 | |
| 433 | |
| 434 | def _clip(pretrained=False, |
nothing calls this directly
no outgoing calls
no test coverage detected