Because the module might be set after the initial step, load the optimizer and scheduler at the start of the training.
EngineModule.initialize_optimizer()
dgs.models.engine.engine.EngineModule.get_target
dgs.models.engine.engine.EngineModule.load_model