EfficientNet模型測試

TPU是什麼?

GPU — Graphics Processing Unit, 圖形處理器,採用多線程SIMD架構,爲圖形處理而生。
CPU — Central Processing Unit 中央處理器, 目前PC core的主流產品。
TPU — Tensor Processing Unit 張量處理器, Google 公司推出的加速人工智能算法的專用處理器。目前一代TPU面向Inference,二代面向訓練。是針對谷歌的深度學習框架TensorFlow專門定製的芯片

官方代碼下載鏈接:

https://github.com/tensorflow/tpu

完整路徑鏈接:

https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet

代碼可以在GPU和TPU上進行訓練。
本人測試環境爲Tensorflow1.13.1, GPU版本。
外部調用,僅需要efficientnet_builder.py,efficientnet_model.py,utils.py和condconv文件夾,其餘的缺啥補啥。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章