GPU代码修改成TPU代码

代码：

https://github.com/appleyuchi/TPU_GPU_Code/blob/master/colab上使用GPU和TPU比较.ipynb

这个代码是GPU和TPU的运行性能比较。
用文本比较网站看了下代码的异同，总共有两处：
－－－－－－－－－－－－－－－－－－－－－第1处－－－－－－－－－－－－－－－－－－－－－－－－－－－－

GPU:

model.compile(optimizer=tf.train.AdamOptimizer(learning_rate=1e-3, ),loss=tf.keras.losses.sparse_categorical_crossentropy,metrics=['sparse_categorical_accuracy']
)

TPU:

import os
tpu_model = tf.contrib.tpu.keras_to_tpu_model(model,strategy=tf.contrib.tpu.TPUDistributionStrategy(tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR']))
)
tpu_model.compile(optimizer=tf.train.AdamOptimizer(learning_rate=1e-3, ),loss=tf.keras.losses.sparse_categorical_crossentropy,metrics=['sparse_categorical_accuracy']
)

－－－－－－－－－－－－－－－－－－－－－－－－－第2处－－－－－－－－－－－－－－－－－－－－－－－－－
GPU:

%%time model.fit_generator(train_gen(1024), epochs=5, steps_per_epoch=100, validation_data=(x_test, y_test))

TPU:

%%time
tpu_model.fit_generator(train_gen(1024), epochs=5, steps_per_epoch=100, validation_data=(x_test, y_test))

GPU代码修改成TPU代码相关推荐

python代码转成php代码的工具或者go转成php的代码，想把odoo改成成php swoole当成web服务+go的架构
目前市场上有一些可以将Python代码转换为PHP代码的工具,例如: Transcrypt:Transcrypt是一个将Python代码转换为JavaScript和PHP的工具.它可以将Python代 ...
怎样从php转向java_php代码转成java代码
hashmap = new hashmap(){ { put("appid", "123"); put("apikey", "45 ...
如何一键将 Vue2 代码转成 Vue3 代码
作者:阿里妈妈前端快爆 https://juejin.cn/post/6977259197566517284 Vue3 已经出来有一段时间了,很多朋友早已熟读了文档,写了好几个 Demo,馋 Comp ...
hp代码改成linux代码,HP Database Archiving Software远程任意代码执行漏洞
发布日期:2011-12-29 更新日期:2011-12-31 受影响系统: HP HP Database Archiving Software 6.31 描述: ------------------ ...
python3代码转python2_Python2代码转成Python3代码
1.利用anaconda软件自带的功能: 1.找2to3.py文件我的anaconda装在了D盘下的Anaconda文件夹下你需要找到anaconda下的script文件夹里面的2to3-scri ...
python2转python3代码_Python2代码转成Python3代码
1.利用anaconda软件自带的功能: 1.找2to3.py文件我的anaconda装在了D盘下的Anaconda文件夹下你需要找到anaconda下的script文件夹里面的2to3-scri ...
Java库转oc_急急急！各位大神：一段JAVA代码转成OC代码。
该楼层疑似违规已被系统折叠隐藏此楼查看此楼全局定义 private static final String ENCODE_TYPE = "UTF-8"; private sta ...
tensorflow代码翻译成pytorch代码 -详细教程+案例
如何把Kotlin代码转成Java代码、把Java代码转成Kotlin代码
Java转Kotlin 打开File –> Settings –> Plugins选项,输入"Kotlin"搜索插件,安装成功重启Android Studio 选中需要 ...

GPU代码修改成TPU代码

GPU代码修改成TPU代码相关推荐

最新文章

热门文章