Merge pull request #2296 from yjmade/patch-1

Fix `Segmentation Fault` when loading very big corpus file

Merge pull request #2296 from yjmade/patch-1
Fix `Segmentation Fault` when loading very big corpus file
32105a38 · Andrew M Dai · GitHub · 0f97888b · 0375c800 · 32105a38
Commit 32105a38 authored Sep 18, 2017 by Andrew M Dai Committed by GitHub Sep 18, 2017
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tutorials/embedding/word2vec_kernels.cc tutorials/embedding/word2vec_kernels.cc +1 -1

No files found.
--- a/tutorials/embedding/word2vec_kernels.cc
+++ b/tutorials/embedding/word2vec_kernels.cc
@@ -128,7 +128,7 @@ class SkipgramWord2vecOp : public OpKernel {
  random::SimplePhilox rng_ GUARDED_BY(mu_);
  int32 current_epoch_ GUARDED_BY(mu_) = -1;
  int64 total_words_processed_ GUARDED_BY(mu_) = 0;
-  int32 example_pos_ GUARDED_BY(mu_);
+  int64 example_pos_ GUARDED_BY(mu_);
  int32 label_pos_ GUARDED_BY(mu_);
  int32 label_limit_ GUARDED_BY(mu_);