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文件夹,其余的缺啥补啥。

你可能感兴趣的:(网络结构)