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