浏览代码

add infer code

often 5 月之前
父节点
当前提交
6e709ad61d
共有 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

@@ -244,7 +244,7 @@ class Main(object):
             init_model_path,
             mode=int(model_mode))
         epoch_start_time = time.time()
-
+        epoch = 0
         if sync_mode == "heter":
             self.heter_train_loop(epoch)
         elif reader_type == "QueueDataset":