From ffb1a2250d42766b5d8d42c67ac957754ad83628 Mon Sep 17 00:00:00 2001
From: LielinJiang <50691816+LielinJiang@users.noreply.github.com>
Date: Tue, 5 Jan 2021 10:59:29 +0800
Subject: [PATCH] fix load inference model (#139)

---
 ppgan/apps/base_predictor.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ppgan/apps/base_predictor.py b/ppgan/apps/base_predictor.py
index 388f841..2cdd3e3 100644
--- a/ppgan/apps/base_predictor.py
+++ b/ppgan/apps/base_predictor.py
@@ -38,11 +38,10 @@ class BasePredictor(object):
                     param_file = file_name
 
             self.program, self.feed_names, self.fetch_targets = paddle.static.load_inference_model(
-                dirname=self.weight_path,
+                self.weight_path,
                 executor=self.exe,
                 model_filename=model_file,
                 params_filename=param_file)
-            print(self.feed_names)
 
     def base_forward(self, inputs):
         if paddle.in_dynamic_mode():
-- 
GitLab