丁云鹏 преди 6 месеца
родител
ревизия
ca02c3ae79
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

+ 1 - 1
recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

@@ -185,7 +185,7 @@ class Main(object):
 
         for epoch in range(epochs):
             fleet.load_inference_model(
-                os.path.join(init_model_path, str(epoch)),
+                init_model_path,
                 mode=int(model_mode))
             epoch_start_time = time.time()