
工具
忘泪
Github:https://github.com/wangleihitcs
Email:wangleihitcs@qq.com
展开
-
Ubuntu16.04 + Cuda-9.0 + Cudnn-7.1.4 + TensorFlow1.8(极其简单)
步骤 1.Ubuntu16.04 LTS 2.配置Nvidia显卡驱动 3.Cuda-9.0 4.Cudnn-7.1.4 5.TensorFlow1.8 1. Ubuntu16.04 LTS 安装Ubuntu16.04不详细说明,网上很多博客 2. 配置Nvidia显卡驱动 网上很多博客,但是我介绍一种简单的安装方法,找到"系统设置"->“软件与更新”-&原创 2018-11-15 16:39:52 · 8104 阅读 · 0 评论 -
Tensorflow 更改模型保存的ckpt文件中的变量名
简介 因为需要用到多个pre-train的CNN模型,想到一个更改ckpt文件变量名的方法,这样只需在代码中更改scope名即可。 核心代码 以inception v3举例,inception_v3.ckpt下载 import tensorflow as tf def rename_var(ckpt_path, new_ckpt_path): with tf.Session() as s...原创 2019-04-12 22:32:51 · 2967 阅读 · 0 评论