站点信息
站点链接:https://deeplearning4j.konduit.ai/
站点标题:基于JVM 的深度学习框架
收录时间:2023-11-09 16:26:54
访问次数:258次
站点关键词:DL4J
DL4J(Deeplearning4j)是Skymind开源并维护的一个基于JVM 的深度学习框架。也是在Spark平台上可以原生支持分布式模型训练的为数不多的框架之一。DL4J还支持多GPU集群,可以与高性能异构计算框架无缝衔接,进一步提升运算性能。
DL4J中除了ND4J外,还有DataVec(用于数据集的加载和转换),SameDiff(基于图的自动微分、深度学习框架),Arbiter(超参数搜索),LibND4J(底层C++库,调用cuDNN、OpenBLAS等库支持CPU/GPU计算)。ND4J通过这些库共同支撑基于JVM的深度学习应用程序的所有需求。
DL4J将数据集处理与算法模型训练分开处理,使用DataVec库加载和转换数据集,使用张量和 ND4J库训练模型。通过DataVec库的RecordReaderDataSetIterator获取数据,ND4J库的DataSet(存储数据特征和标签的容器)对训练数据进行封装,可以封装单条训练数据,也可以封装一个 Mini-batch,每一次迭代用一个DataSet更新神经网络中参数。在DL4J中构建训练数据集,最终就是生成一个DataSet迭代器或DataSet序列。
温馨提示:当前网站仅在本站做收录展示,同时在2023年11月09日入库时,该网站内容都属于安全正规,但不能保证一直都是安全正规,请注意自行判断站点的安全性,如有内容出现违规,可通过本站站点反馈功能进行反馈,或联系站长进行删除,以免财产损失!
站点截图
相关推荐
最新收录
- 暂无新站点加入