Restore original forward method
(self)
| 55 | self.is_active = True |
| 56 | |
| 57 | def disable(self): |
| 58 | """Restore original forward method""" |
| 59 | if not self.is_active: |
| 60 | return |
| 61 | |
| 62 | # Restore original forward method |
| 63 | self.model.forward = self.original_forward |
| 64 | self.is_active = False |
| 65 | |
| 66 | |
| 67 | class MotionVarianceOptimizer: |
no outgoing calls
no test coverage detected