Slim在CPU上跑的方法

原创于 2018-06-14 14:33:57 发布 · 1k 阅读

·

2

·

CC 4.0 BY-SA版权

文章标签：

64 篇文章

订阅专栏

本文介绍如何在TensorFlow的train_image_classifier中调整'clone_on_cpu'参数，将部署克隆的设备从默认的GPU切换到CPU，以适应不同的计算环境。

修改train_image_classifier中的这一句

tf.app.flags.DEFINE_boolean('clone_on_cpu', True,
                            'Use CPUs to deploy clones.')

参数的False改为True即可

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。