diff --git a/core/trainers/single_trainer.py b/core/trainers/single_trainer.py
index ce1aa987de70aa56b41814f4d9e57185126c3ce9..dd06b0b04ed6f12e946064e7658fc199a5390b90 100755
--- a/core/trainers/single_trainer.py
+++ b/core/trainers/single_trainer.py
@@ -32,7 +32,7 @@ logger = logging.getLogger("fluid")
 logger.setLevel(logging.INFO)
 
 
-class SingleTrainerYamlOpt(TranspileTrainer):
+class SingleTrainer(TranspileTrainer):
     def __init__(self, config=None):
         super(TranspileTrainer, self).__init__(config)
         self._env = self._config