提交 5777d636 编写于 作者: X xjqbest

fix

上级 a987bb78
...@@ -112,7 +112,8 @@ class SingleTrainer(TranspileTrainer): ...@@ -112,7 +112,8 @@ class SingleTrainer(TranspileTrainer):
if sparse_slots is None and dense_slots is None: if sparse_slots is None and dense_slots is None:
reader = dataloader_instance.dataloader_by_name( reader = dataloader_instance.dataloader_by_name(
reader_class, dataset_name, self._config_yaml) 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) reader_ins = reader_class(self._config_yaml)
else: else:
reader = dataloader_instance.slotdataloader_by_name( reader = dataloader_instance.slotdataloader_by_name(
...@@ -178,8 +179,8 @@ class SingleTrainer(TranspileTrainer): ...@@ -178,8 +179,8 @@ class SingleTrainer(TranspileTrainer):
model.net(model._infer_data_var, True) model.net(model._infer_data_var, True)
else: else:
model.net(model._data_var, False) model.net(model._data_var, False)
optimizer = model._build_optimizer(opt_name, opt_lr, optimizer = model._build_optimizer(
opt_strategy) opt_name, opt_lr, opt_strategy)
optimizer.minimize(model._cost) optimizer.minimize(model._cost)
model_dict["is_infer"] = is_infer model_dict["is_infer"] = is_infer
self._model[model_dict["name"]][0] = train_program self._model[model_dict["name"]][0] = train_program
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册