diff --git a/core/trainers/single_trainer.py b/core/trainers/single_trainer.py index a09a87b3bae91344d14a032dda0567bf1caf71e2..a6c21265e1a04c89bf0d6cff5302b4c75222ba80 100755 --- a/core/trainers/single_trainer.py +++ b/core/trainers/single_trainer.py @@ -112,7 +112,8 @@ class SingleTrainer(TranspileTrainer): if sparse_slots is None and dense_slots is None: reader = dataloader_instance.dataloader_by_name( reader_class, dataset_name, self._config_yaml) - reader_class = envs.lazy_instance_by_fliename(reader_class, "TrainReader") + reader_class = envs.lazy_instance_by_fliename(reader_class, + "TrainReader") reader_ins = reader_class(self._config_yaml) else: reader = dataloader_instance.slotdataloader_by_name( @@ -178,8 +179,8 @@ class SingleTrainer(TranspileTrainer): model.net(model._infer_data_var, True) else: model.net(model._data_var, False) - optimizer = model._build_optimizer(opt_name, opt_lr, - opt_strategy) + optimizer = model._build_optimizer( + opt_name, opt_lr, opt_strategy) optimizer.minimize(model._cost) model_dict["is_infer"] = is_infer self._model[model_dict["name"]][0] = train_program