Support mini batch in training (facebookresearch#190)

* batch negs Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: Blame Revision: * fix
dna2fork · Jul 19, 2019 · ed40f8b · ed40f8b
1 parent 64db348
commit ed40f8b
Show file tree

Hide file tree

Showing 6 changed files with 293 additions and 244 deletions.
diff --git a/examples/classification_ag_news.sh b/examples/classification_ag_news.sh
@@ -55,7 +55,6 @@ echo "Start to train on ag_news data:"
   -thread 20 \
   -dim 10 \
   -negSearchLimit 5 \
-  -maxNegSamples 3 \
   -trainMode 0 \
   -label "__label__" \
   -similarity "dot" \