tensorflow
扎个冲天揪
冲冲冲冲冲冲鸭
展开
-
维度解读
部分代码显示:其中WC1的维度为[1,20,9,32]x的维度为(?,128 , 1 , 9 )意义为(batch_size, img_height , img_weight , in_channel)w的维度设置(1,20 , 9 , 32),意义为(filter_height, filter_width , in_channels , out_channels),卷...原创 2018-10-26 20:51:55 · 324 阅读 · 0 评论 -
failed to create cublas handle:CUBLAS_STATUS_ALLOC_FAILED
看了很多博客,有些说的是加上gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))这两句,加了并没有用,之后发现......是因为之前的debug窗口没关,占用了GPU。哇...原创 2018-10-27 16:43:55 · 7331 阅读 · 5 评论 -
TensorFlow保存训练模型为pb文件,读取pb恢复模型,C++调用pb模型
【转载】https://blog.csdn.net/sxlsxl119/article/details/81937815 一,第一步,训练保存模型 贴自己源代码太复杂,贴个简化版,表明主要意思就行,别人做的东西不可能和你的完全一样,需要在理解别人意思的基础上,对自己的代码加以更改。 注意看下面代码中的注释!import...转载 2019-07-27 10:38:54 · 222 阅读 · 0 评论