|
@@ -225,7 +225,7 @@ class Main(object):
|
|
|
fleet.save_inference_model(
|
|
|
self.exe, model_dir,
|
|
|
[feed.name for feed in self.inference_feed_var],
|
|
|
- self.inference_target_var
|
|
|
+ self.inference_target_var,
|
|
|
export_for_deployment=True )
|
|
|
else:
|
|
|
paddle.static.save_inference_model(
|