Method initialize

tests/test_deepspeed_training_worker.py:53–57

(self, **kwargs)

        self.engines = []

    def initialize(self, **kwargs):
        self.initialize_calls.append(kwargs)
        engine = FakeDeepSpeedEngine(kwargs["model"], kwargs["optimizer"])
        self.engines.append(engine)
        return engine, kwargs["optimizer"], None, None


class FakeGatheredParameters: