Skip to content

this is a high performance cuda porting of cbow model of word2vec

Notifications You must be signed in to change notification settings

laurencecao/word2vec_cbow

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

word2vec_cbow

this is a high performance cuda porting of cbow model of word2vec

I tested the code on K40 with Intel E5-2650 CPU, its performance is about 15xx K words/thread/sec, it is about 20x faster as cpu version run with 16 threads. AFAIK, this is the fastest implementation.

About

this is a high performance cuda porting of cbow model of word2vec

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Cuda 98.7%
  • Other 1.3%