fixed cub path

Author: jcanny

jni/src/configure

@@ -107,7 +107,7 @@ elif [ "$OS" = "linux" ] ; then
     JAVA_HOME="/usr/java/default"
     fi
     CUDA_HOME="/usr/local/cuda"
-    CUB_HOME=../../../cub
+    CUB_ROOT=../../../cub
     CC="icc"
     GCC="g++"
     NVCC="nvcc"

scripts/testword2vecp.ssc

@@ -7,7 +7,7 @@ val (nn, opts) = Word2Vec.learnPar(mdir+"data%03d.imat.lz4");
 opts.nstart = 0;
 opts.nend = 7;
 opts.npasses = 1;
-opts.batchSize = 100000;
+opts.batchSize = 1000000;
 opts.lrate = 1e-3f;
 opts.vexp = 0.5f
 opts.nreuse = 5