From 34855d8ec3a7746a888cf4fb0385ad9b795163fa Mon Sep 17 00:00:00 2001
From: wuzewu <wuzewu@baidu.com>
Date: Thu, 28 Mar 2019 20:08:35 +0800
Subject: [PATCH] add inference program

---
 paddle_hub/finetune/task.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddle_hub/finetune/task.py b/paddle_hub/finetune/task.py
index 1d75b666..9275e95d 100644
--- a/paddle_hub/finetune/task.py
+++ b/paddle_hub/finetune/task.py
@@ -30,6 +30,7 @@ class Task(object):
         self.graph_var_dict = graph_var_dict
         self._main_program = main_program
         self._startup_program = startup_program
+        self._inference_program = main_program.clone(for_test=True)
 
     def variable(self, var_name):
         if var_name in self.graph_var_dict:
@@ -42,3 +43,6 @@ class Task(object):
 
     def startup_program(self):
         return self._startup_program
+
+    def inference_program(self):
+        return self._inference_program
-- 
GitLab