add constants.py

Author: js1010

cusim/constants.py

@@ -0,0 +1,10 @@
+# Copyright (c) 2021 Jisang Yoon
+# All rights reserved.
+#
+# This source code is licensed under the Apache 2.0 license found in the
+# LICENSE file in the root directory of this source tree.
+
+# pylint: disable=no-name-in-module,too-few-public-methods,no-member
+
+EPS = 1e-10
+WARP_SIZE = 32

cusim/culda/pyculda.py

@@ -18,9 +18,8 @@
 from cusim import aux, IoUtils
 from cusim.culda.culda_bind import CuLDABind
 from cusim.config_pb2 import CuLDAConfigProto
+from cusim.constants import EPS, WARP_SIZE
 
-EPS = 1e-10
-WARP_SIZE = 32
 
 class CuLDA:
   def __init__(self, opt=None):

cusim/cuw2v/pycuw2v.py

@@ -17,9 +17,7 @@
 from cusim import aux, IoUtils
 from cusim.cuw2v.cuw2v_bind import CuW2VBind
 from cusim.config_pb2 import CuW2VConfigProto
-
-EPS = 1e-10
-WARP_SIZE = 32
+from cusim.constants import EPS, WARP_SIZE
 
 class CuW2V:
   def __init__(self, opt=None):
@@ -102,7 +100,7 @@ def train_model(self):
     for epoch in range(1, self.opt.epochs + 1):
       self.logger.info("Epoch %d / %d", epoch, self.opt.epochs)
       self._train_epoch(h5f)
-    self.pull()
+    self.obj.pull()
     h5f.close()
 
   def _train_epoch(self, h5f):

examples/example1.py

@@ -68,7 +68,7 @@ def run_lda():
 
 def run_w2v():
   opt = {
-    "c_log_level": 3,
+    # "c_log_level": 3,
     "data_path": DATA_PATH,
     "processed_data_dir": PROCESSED_DATA_DIR,
     # "skip_preprocess":True,