問題タブ [google-cloud-tpu]

質問する

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

171 問題
Newest
Active
Bountied
318
Unanswered
0 投票する
1 に答える
2797 参照

python - Colab で keras_to_tpu_model を使用すると、TPU が CPU と同じくらい遅く実行される

私はtf.contrib.tpu.keras_to_tpu_model自分のコードを TPU で実行できるようにしていましたが、エポックを完了するのに 170 時間かかりましたが、CPU には同じ時間がかかり、GPU にはエポックあたり 40 時間しかかかりませんでした。バッチサイズを調整しようとしましたが、何も変わりませんでした。 GPUで実行すると、入力関数が実行時間の20％を占める可能性があることをテストしたので、それが主な理由ではないかもしれません.

ここに私のコードがあります: https://github.com/WangHexie/DHNE/blob/master/src/hypergraph_embedding.py

colab で実行:

  1. TPU: https://colab.research.google.com/gist/WangHexie/30c385509f9cd93be747f04c39f039a4/tpu-error.ipynb
  2. GPU：<a href="https://colab.research.google.com/gist/WangHexie/5bfac53bf92ef0ad527f15ddbf8705e1/-gpu-ipynb.ipynb" rel="nofollow noreferrer">https://colab.research.google.com /gist/WangHexie/5bfac53bf92ef0ad527f15ddbf8705e1/-gpu-ipynb.ipynb

モデル:


12345678910