l8l技術メモ

機械学習・深層学習・組込みとかのメモ

Tensorflow(Keras)かPytorchか

KerasかPytorchか

 

簡単に調べた・知っている範囲だと

■ Keras

変なことしないなら少ない行数ですむ(?)

Tensorflowバックエンドがついてくる(というかTF2.0でKerasと融合しつつある)

TFバックエンドなら、モバイル(TFLite)への移行がスムーズ。EdgeTPUがついてくる。

■ Pytorch

細かいところもいじりたいなら最終的に楽(?)

Kerasより早い

 

[ 2019/11/01追記]

以下記事によると、2019年現在10月現在研究用途ではPytorchが圧倒しつつある。Production用途では、Tensorflowが依然優勢。現状experimentalではあるがpytorchのmobile対応が進んでいる。筆者的にはpytorch推し。

 https://thegradient.pub/state-of-ml-frameworks-2019-pytorch-dominates-research-tensorflow-dominates-industry/