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文件夾,其餘的缺啥補啥。