dgs.models.engine.engine.EngineModule.set_model_mode

EngineModule.set_model_mode(mode: str) None[source]

Set model mode to train or test.