Home Explore Help
Sign In
algorithm
/
recommend-model
15
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

add infer code

often 5 months ago
parent
e88df21fd8
commit
6e709ad61d
1 changed files with 1 additions and 1 deletions
Unified View Show Diff Stats
  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
View File 

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