(self)
| 409 | |
| 410 | |
| 411 | def remove_weight_norm(self): |
| 412 | def _remove_weight_norm(m): |
| 413 | try: |
| 414 | print(m) |
| 415 | nn.utils.remove_weight_norm(m) |
| 416 | except ValueError: # this module didn't have weight norm |
| 417 | return |
| 418 | |
| 419 | self.apply(_remove_weight_norm) |
| 420 | |
| 421 | |
| 422 |