|
|
@@ -301,8 +301,8 @@ class Main(object):
|
|
|
output_file = os.path.join(output_dir, f"epoch_{epoch}_results.jsonl")
|
|
|
|
|
|
# 创建处理器实例
|
|
|
- fetch_handler = InferenceFetchHandler(output_file)
|
|
|
- fetch_handler.set_var_dict(self.metrics)
|
|
|
+ fetch_handler = InferenceFetchHandler(var_dict = self.metrics, output_file =output_file)
|
|
|
+ # fetch_handler.set_var_dict(self.metrics)
|
|
|
|
|
|
print(paddle.static.default_main_program()._fleet_opt)
|
|
|
self.exe.infer_from_dataset(
|
丁云鹏