-
NETLAB是一款以C++编写的神经网络工具,其核心功能是记录序列数据,并重点关注长短期记忆(LSTM)层和受限玻尔兹曼机(RBM)层的运用
资源介绍
网络实验室
一个C ++神经网络主要记录序列,主要关注长短期记忆(LSTM)和受限Boltsmann机器(RBM)层。 随附论文供参考:[使用LSTM-RTRBM的和声音乐建模],ACM多媒体,2015年。
创建者和维护者:齐柳
特征
易于使用Matlab之类的语法创建新层(只需要预先分配内存)
层之间的时间自动反向传播(BPTT)计算
统一的CPU / GPU代码:可以同时在CPU和GPU中运行
可扩展:易于使用Google的protobuf配置net参数
重量快照,易于加载模型和可视化
依存关系
CUDA库版本6.5(推荐),6.0、5.5或5.0,以及CUDA 6或319. *的最新驱动程序版本(对于CUDA 5(而非331. *))
BLAS(通过ATLAS,MKL或OpenBLAS提供)
提升(> = 1.55)
mshadow( )
google-glog,协议缓