深度学习实操
深夜虫鸣
这个作者很懒,什么都没留下…
展开
-
keras多显卡训练
使用keras进行训练,默认使用单显卡,即使设置了os.environ['CUDA_VISIBLE_DEVICES']为两张显卡,也只是占满了显存,再设置tf.GPUOptions(allow_growth=True)之后可以清楚看到,只占用了第一张显卡,第二张显卡完全没用。要使用多张显卡,需要按如下步骤:(1)import multi_gpu_model函数:from keras.utils ...原创 2018-12-04 10:25:34 · 3806 阅读 · 6 评论 -
tensorflow查看ckpt各节点名称
运行下列脚本,可以打印出模型各个节点变量的名称:from tensorflow.python import pywrap_tensorflowimport oscheckpoint_path=os.path.join('model.ckpt-131805')reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)var_...原创 2019-01-07 10:38:13 · 17231 阅读 · 7 评论