tensorflow - GOOGLE COLAB TPU を使用して IMAGENET で VGG-16 モデルをトレーニングするにはどのくらいの時間がかかりますか?

Question

興味があるのですが、GOOGLE COLAB TPU を使用して IMAGENET で VGG16 モデルをトレーニングするにはどのくらいの時間がかかりますか? 誰かが答えにたどり着くために行った計算を説明できれば、それは素晴らしいことです!

Answer 1

モデル e2e のトレーニングにかかる​​時間を正確に見積もることは非常に困難です。しかし、非常に大まかな見積もりを探しているだけだと仮定すると、( code )持っているこの ResNet50 実装は、v2- 8 TPU デバイス。VGG16 がステップ時間 ( https://github.com/jcjohnson/cnn-benchmarks#cnn-benchmarks ) で十分に近いことを考えると、収束もそれに比例すると予想されます。ただし、これは非常に大まかな見積もりであり、実際のパフォーマンスは、実装がどの程度最適化されているかによっても異なります。