(args)
| 33 | self.reset_idx = [arg_names.index(k) for k in reset_to_zero] |
| 34 | |
| 35 | def _hook(args): |
| 36 | for i in self.reset_idx: |
| 37 | args[i].zero_() |
| 38 | self.hook = _hook |
| 39 | self.arg_names = arg_names |
| 40 | # prune configs |
nothing calls this directly
no outgoing calls
no test coverage detected