PyTorch
优点:
1、非常灵活
2、高水平的速度和效率
3、提供大量预先训练的模型
缺点:
1、文件不清楚
2、它基于一种不流行的语言Lua
Keras
优点:
1、使用方便
2、容易扩展
3、在CPU和GPU上无缝运行
4、与其他框架无缝协作
缺点:
1、无法有效地用作独立框架
Caffe
优点:
1、源码简单易读
2、容易扩展
3、很好的效率
缺点:
1、对循环网络和语言建模的支持薄弱
Theano
优点:
1、运行效率高
2、在CPU和GPU上无缝运行
缺点:
1、调试错误比较困难
2、大型模型有较长的编译时间
Cognitive Toolkit
优点:
1、非常灵活
2、速度快
3、允许分布式训练
4、支持C++、C#、Java、Python
缺点:
1、缺乏可视化
2、源码可读性比较差
TensorFlow
优点:
1、简单易学
2、可视化工具
3、允许分布式训练
4、在CPU和GPU上无缝运行
5、支持JavaScript、Java、C++、Python
缺点:
1、比其它框架稍慢
mxNet
(缺)
这么多框架选那个好呢?小孩子才需要做选择
DJL:我全要
DJL框架介绍
djl基于java强大的抽象能力,将各ai框架抽象成一个一个的算法引擎,类似于在java中操作数据库都可以通过jdbc一样,对各种ai引擎提供了统一的访问接口,可以方便的部署和推理,同时也可以用于训练