Skip to content

Commit 32105a3

Browse files
authored
Merge pull request tensorflow#2296 from yjmade/patch-1
Fix `Segmentation Fault` when loading very big corpus file
2 parents 0f97888 + 0375c80 commit 32105a3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tutorials/embedding/word2vec_kernels.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ class SkipgramWord2vecOp : public OpKernel {
128128
random::SimplePhilox rng_ GUARDED_BY(mu_);
129129
int32 current_epoch_ GUARDED_BY(mu_) = -1;
130130
int64 total_words_processed_ GUARDED_BY(mu_) = 0;
131-
int32 example_pos_ GUARDED_BY(mu_);
131+
int64 example_pos_ GUARDED_BY(mu_);
132132
int32 label_pos_ GUARDED_BY(mu_);
133133
int32 label_limit_ GUARDED_BY(mu_);
134134

0 commit comments

Comments
 (0)