丁云鹏 5 månader sedan
förälder
incheckning
ca02c3ae79
1 ändrade filer med 1 tillägg och 1 borttagningar
  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()