【文件属性】:
文件名称:java二叉树源码-cuda-word2vec:CBOW模型的cuda实现(word2vec)
文件大小:26KB
文件格式:ZIP
更新时间:2021-06-05 07:57:10
系统开源
java二叉树源码cuda-word2vec
CBOW模型的cuda实现
特征
与
Nvidia
GPU
并行加速
对于任意大的训练集需要恒定的内存
流实现不需要训练数据随机访问
自动验证并在训练期间显示验证数据负对数似然
支持自定义词二叉树构造。
使用由任何其他无监督模型构建的自己的二叉树来提高模型性能
更新
0.1
利用局部性优化GPU内存IO效率
使用
cuda
纹理优化内存性能
通过训练数据的二进制预处理优化磁盘到内存的
IO
效率并节省解析开销
反向传播梯度检查测试完成
实现自动
cuda
内存管理实用程序来管理
GPU
资源
要求
CUDA
5.5
cuda
推力
nvcc
编译器
1GB
或更多
GPU
内存(推荐
2GB+)
汇编
Linux:如果
nvcc
命令可用,则使用
makefile
进行编译
Windows:使用
-arch=compute_30
和
-code=sm_30
手动编译源文件
源文件:language_network_main.cu
language_network_utility.cu
language_network_kernel.cu
i
【文件预览】:
cuda-word2vec-master
----src()
--------language_network_utility.cu(10KB)
--------language_network_test.cu(11KB)
--------language_network_main.cu(5KB)
--------language_network_kernel.cu(11KB)
--------includes()
----LICENSE(1KB)
----README.md(3KB)
----makefile(505B)