|
@@ -243,7 +243,7 @@ class Main(object):
|
|
[self.inference_target_var], self.exe)
|
|
[self.inference_target_var], self.exe)
|
|
|
|
|
|
if fleet.is_first_worker():
|
|
if fleet.is_first_worker():
|
|
- model_dir = "{}/{}".format(save_model_path, epochs)
|
|
|
|
|
|
+ model_dir = "{}/{}".format(save_model_path, epochs - 1)
|
|
oss_object_name = self.config.get("runner.oss_object_name", "dyp/dnn_plugin_new.tar.gz")
|
|
oss_object_name = self.config.get("runner.oss_object_name", "dyp/dnn_plugin_new.tar.gz")
|
|
# trans to new format
|
|
# trans to new format
|
|
# {"model_filename":"", "params_filename":""} fleet每个参数一个文件,需要同这种方式加载
|
|
# {"model_filename":"", "params_filename":""} fleet每个参数一个文件,需要同这种方式加载
|