Dcurry-CSDN博客

原创深度学习Tensorflow的坑（二：训练一个epoch卡住不动，程序不终止也不继续）

第一篇承接上文在搭载好环境之后，满心欢喜准备搭载模型的绘图（损失曲线、验证精度）功能，我就先使用10个epoch跑一下，结果。它卡在199/200了！之前从没遇到过这个情况。先看控制台报错，说是导入CUDNN的DLL文件不成功，得，又得复制粘贴了。接着跑，还是不成功。网上很多报错都是pytorch版本得dataloader，不太能够适配。我就开始自身琢磨，我发现训练过程是没问题得，那么问题是出在我的valmodel，最后一步savemodel根本没走过去（QAQ深度学习不会打断点啊！）。那我的

2021-11-22 10:54:45 13690

原创深度学习Tensorflow的坑（一：无法使用GPU加速训练深度学习模型，CUDA与CUDNN相关）

项目场景：本研究僧为了秋招卷大厂放弃了半年的深度学习，重回Python的一些坑记录问题与解决：1：版本问题想用GPU加速深度学习的训练一定不能同时下载tensorflow和tensorflow-gpu。如果遇到了什么tensorflow找不到的bug，请直接删除所有带有tensorflow的包，然后找到对应版本的下载对应版本是指CUDNN和CUDA以及你自身显卡实力对应的tensorflow-gpu版本。2.CUDA和CUDNN下载（2、3一起看食用体验更佳！）CUDA下载比较简单CU

2021-11-21 17:00:17 3854

原创 ‘Model‘ object has no attribute ‘_container_nodes‘

解决方案：如果使用的是keras ，请保留if node_key in model._container_nodes:如何是从tf.keras.leyers 导出的：那么按如下修改Fix to tensorflow#17633（https://github.com/tensorflow/tensorflow/issues/17633）‘Model’ object has no attribute ‘_container_nodes’ error when using tf.keras.uti

2020-11-08 15:49:30 452 1

原创实现图像三通道转为单通道的方法

项目场景：from tensorflow.keras.preprocessing.image import img_to_array, load_imgimg = load_img(img, target_size=(48, 48), color_mode=“grayscale”)img = img_to_array(img) # 灰度化

2020-10-30 10:56:34 4020 1

原创 tensorflow官方github 预训练模型下载

tensorflow官方github 预训练模型下载一、官网链接二、使用步骤1.寻找你想要的预训练模型2.下载对应文件Code对应程序源码Checkpoint对应预训练模型提示一、官网链接示例：https://github.com/tensorflow/models/tree/master/research/slim二、使用步骤1.寻找你想要的预训练模型图示如下：2.下载对应文件Code对应程序源码Checkpoint对应预训练模型提示1、一般都是用imagenet上的预训练模型.

2020-10-18 14:06:02 2864 1

原创 Tensorflow中tf.app.flags和tf.app.run。

简记tf.app.flags正文一、tf.app.flags和tf.app.run。二、简记标题总结正文一、tf.app.flags和tf.app.run。经常在程序中看到有tf.app.flags和tf.app.run字样的代码，这两段代码究竟是什么作用，我在这里简单记录一下。希望大神批评指正。二、简记首先我们通过tf.app.flags来调用这个flags.py文件，这样我们就可以用flags.DEFINE_interger/float()来添加命令行参数，而FLAGS=flags.FLAG

2020-10-15 14:21:18 292 1

原创 win10 python3 安装 Tensorflow-gpu 问题及解决

这里写自定义目录标题准备工作： win10 下， python3.6安装Tensorflow-gpu 2.0.0完成.准备工作： win10 下， python3.6安装Tensorflow-gpu 2.0.0完成.一开始直接用pycharm 的setting下载tensorflow2.0.0版本，出错。网络上建议我更新pip和setuptools，然而这样，我就选择用pip install tensorflow，结果还是不行，第一下载太慢，第二最后还是会提醒出错。于是我在考虑是否是版本的原因。我

2020-09-16 10:25:16 381

Dcurry的博客