タグ

ブックマーク / intheweb.io (1)

  • TensorFlow でのデータ入力方法まとめ ①

    あゆたディープラーニング担当の佐々木です。 今日は TensorFlow でモデルにデータを入れる手法についてまとめてみます。 パフォーマンスに関する家の記事曰く、 チュートリアル等々で用いられている定番の feed_dict を使う手法だと、モデルの計算量やデータのサイズ次第では GPU を 活かしきれない事態になるそうです。(GPU の使用率がやたら低くなる) 原因としては、 CPU -> GPU 間のデータ転送が遅いことにあるようで、 次のデータが来る前に計算が終わってしまい、GPU が暇を持て余した状態になる場合があるようです。 そこで今回は Queue を用いてデータの転送を行う手法を実装してみます。 といっても先に結論を言ってしまうと、MNIST Expert ではどの手法でも GPU の 使用率は同じくらいだったので、場合によるのかなと。 CNN のようにもともと計算量の多

  • 1