diff options
Diffstat (limited to 'src/layers/word2vec_reader.h')
-rwxr-xr-x | src/layers/word2vec_reader.h | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/src/layers/word2vec_reader.h b/src/layers/word2vec_reader.h index f18fd439..a7e85592 100755 --- a/src/layers/word2vec_reader.h +++ b/src/layers/word2vec_reader.h @@ -18,8 +18,6 @@ public: LOG(info, "[data] Loading embedding vectors from {}", fileName); io::InputFileStream embFile(fileName); - ABORT_IF(!embFile.isOpen(), - "Unable to open file with embeddings: " + fileName); std::string line; std::vector<std::string> values; @@ -75,19 +73,19 @@ private: values.reserve(dimEmb); // Glorot numal distribution float scale = sqrtf(2.0f / (dimVoc + dimEmb)); - + // @TODO: switch to new random generator back-end. - // This is rarly used however. + // This is rarly used however. std::random_device rd; std::mt19937 engine(rd()); - + std::normal_distribution<float> d(0, scale); auto gen = [&d, &engine] () { return d(engine); }; std::generate(values.begin(), values.end(), gen); - + return values; } }; |