Browse Source

Add -verbose option for LR update

Edouard Grave 9 years ago
parent
commit
cd5726ef4b
2 changed files with 2 additions and 2 deletions
  1. 1 1
      src/args.cc
  2. 1 1
      src/fasttext.cc

+ 1 - 1
src/args.cc

@@ -27,7 +27,7 @@ Args::Args() {
   minn = 3;
   minn = 3;
   maxn = 6;
   maxn = 6;
   thread = 12;
   thread = 12;
-  verbose = 1000;
+  verbose = 10000;
   t = 1e-4;
   t = 1e-4;
   label = "__label__";
   label = "__label__";
 }
 }

+ 1 - 1
src/fasttext.cc

@@ -233,7 +233,7 @@ void trainThread(Dictionary& dict, Matrix& input, Matrix& output,
       skipgram(dict, model, line, loss, nexamples);
       skipgram(dict, model, line, loss, nexamples);
     }
     }
 
 
-    if (tokenCount > 10000) {
+    if (tokenCount > args.verbose) {
       info::allWords += tokenCount;
       info::allWords += tokenCount;
       info::allLoss += loss;
       info::allLoss += loss;
       info::allN += nexamples;
       info::allN += nexamples;