|
@@ -260,13 +260,14 @@ class Main(object):
|
|
"dump_fields": config.get("runner.dump_fields")
|
|
"dump_fields": config.get("runner.dump_fields")
|
|
})
|
|
})
|
|
print(paddle.static.default_main_program()._fleet_opt)
|
|
print(paddle.static.default_main_program()._fleet_opt)
|
|
- self.exe.infer_from_dataset(
|
|
|
|
|
|
+ results = self.exe.infer_from_dataset(
|
|
program=paddle.static.default_main_program(),
|
|
program=paddle.static.default_main_program(),
|
|
dataset=self.reader,
|
|
dataset=self.reader,
|
|
fetch_list=fetch_vars,
|
|
fetch_list=fetch_vars,
|
|
fetch_info=fetch_info,
|
|
fetch_info=fetch_info,
|
|
print_period=print_step,
|
|
print_period=print_step,
|
|
debug=debug)
|
|
debug=debug)
|
|
|
|
+ print("results {}".format(results))
|
|
|
|
|
|
|
|
|
|
def heter_train_loop(self, epoch):
|
|
def heter_train_loop(self, epoch):
|