| 46 | |
| 47 | |
| 48 | class FakeDeepSpeedModule: |
| 49 | def __init__(self): |
| 50 | self.initialize_calls = [] |
| 51 | self.engines = [] |
| 52 | |
| 53 | def initialize(self, **kwargs): |
| 54 | self.initialize_calls.append(kwargs) |
| 55 | engine = FakeDeepSpeedEngine(kwargs["model"], kwargs["optimizer"]) |
| 56 | self.engines.append(engine) |
| 57 | return engine, kwargs["optimizer"], None, None |
| 58 | |
| 59 | |
| 60 | class FakeGatheredParameters: |
no outgoing calls