fix load inference model (PaddlePaddle#139)

SlamDunkDcc · Jan 5, 2021 · ffb1a22 · ffb1a22
1 parent f743afd
commit ffb1a22
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ppgan/apps/base_predictor.py b/ppgan/apps/base_predictor.py
@@ -38,11 +38,10 @@ def build_inference_model(self):
                     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():