Tensorflow: Model trained(checkpoint files) on GPU can be converted to CPU running model?

[亡魂溺海] 提交于 2019-12-13 09:48:52

问题


A model is trained with GPU and result is saved by checkpoint file. The saved checkpoint file can be run by cpu-tensorflow? If not, can convert the saved checkpoint file so as to run model in cpu-tensorflow?


回答1:


Yes! It normally can!

The exception is with tf.device('gpu:0') statements. If you do not have them in your code you are good to go!

Good luck!



来源:https://stackoverflow.com/questions/42951543/tensorflow-model-trainedcheckpoint-files-on-gpu-can-be-converted-to-cpu-runni

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!