Set model mode to train or test.
EngineModule.set_model_mode()
dgs.models.engine.engine.EngineModule.save_model
dgs.models.engine.engine.EngineModule.terminate