Browse Source

save model

丁云鹏 5 months ago
parent
commit
3c334a3a4f

+ 3 - 3
recommend-model-produce/src/main/python/tools/static_ps_trainer_v2.py

@@ -268,15 +268,15 @@ class Main(object):
                         feedvars.append(program.global_block().vars[var_name])
 
                 paddle.static.save_inference_model(
-                    os.path.join(model_dir, "dnn_plugin_new"),
+                    os.path.join(model_dir, "dnn_plugin_new/dssm"),
                     feedvars,
                     fetchvars, 
                     self.exe,
                     program = program)
 
-                compress.compress_tar(os.path.join(model_dir, "dnn_plugin_new/dssm"), "dssm.tar.gz")
+                compress.compress_tar(os.path.join(model_dir, "dnn_plugin_new"), "dnn_plugin_new.tar.gz")
                 client = HangZhouOSSClient("art-recommend")
-                client.put_object_from_file("dyp/dssm.tar.gz", "dssm.tar.gz")
+                client.put_object_from_file("dyp/dssm.tar.gz", "dnn_plugin_new.tar.gz")
                 while True:
                     time.sleep(300)
                     continue;