|
@@ -225,7 +225,8 @@ 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_varm
|
|
|
+ export_for_deployment=True )
|
|
|
else:
|
|
|
paddle.static.save_inference_model(
|
|
|
model_dir,
|