dssm train

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

@@ -51,7 +51,12 @@ logger = logging.getLogger(__name__)
 import json
 
 class InferenceFetchHandler(FetchHandler):
-    def __init__(self, output_file, batch_size=1000):
+    def __init__(self, var_dict, period_secs, output_file, batch_size=1000):
+        assert var_dict is not None
+        self.var_dict = var_dict
+        self.period_secs = period_secs
+
+
         self.output_file = output_file
         self.batch_size = batch_size
         self.current_batch = []