エントリーの編集
エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
NHWC vs NCHW on Google Colab - TadaoYamaokaの開発日記
記事へのコメント0件
- 注目コメント
- 新着コメント
このエントリーにコメントしてみましょう。
注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています
- バナー広告なし
- ミュート機能あり
- ダークモード搭載
関連記事
NHWC vs NCHW on Google Colab - TadaoYamaokaの開発日記
畳み込みの入力データの形式には、NHWCとNCHW があるが、どちらがTPUに最適か実験してみた。 TensorFlow... 畳み込みの入力データの形式には、NHWCとNCHW があるが、どちらがTPUに最適か実験してみた。 TensorFlowのデフォルトはNHWCで、ChainerのデフォルトはNCHWになっている。 cuDNNはNCHWに最適化されている。 https://www.tensorflow.org/guide/performance/overview しかし、TensorCoreは、NHWCに最適化されている。 Volta Tensor コア GPU が AI パフォーマンスの新記録を達成 | NVIDIA なお、TensorFlowのCPUモードでは、MaxpoolingはNHWCしかサポートしていない。 TPUではどちらに最適化されているか調べてもわからなかった。 そこで、カラー画像のデータセットであるCIFAR-10を使って、2層の畳み込み層のあるネットワークを学習させて、実際に測ってみた