tensorflow中ckpt转pb模型

参考代码:python import tensorflow as tf import os from tensorflow.python.framework import graph_utilsession def set_config():#设置GPU使用率     # 控制使用率     os.environ['CUDA_VISIBLE_DEVICES'] = '0'     # 假若有16G
相关文章
相关标签/搜索