- 博客(7)
- 收藏
- 关注
原创 深度学习Tensorflow的坑(二:训练一个epoch卡住不动,程序不终止也不继续)
第一篇承接上文在搭载好环境之后,满心欢喜准备搭载模型的绘图(损失曲线、验证精度)功能,我就先使用10个epoch跑一下,结果。它卡在199/200了!之前从没遇到过这个情况。先看控制台报错,说是导入CUDNN的DLL文件不成功,得,又得复制粘贴了。接着跑,还是不成功。网上很多报错都是pytorch版本得dataloader,不太能够适配。我就开始自身琢磨,我发现训练过程是没问题得,那么问题是出在我的valmodel,最后一步savemodel根本没走过去(QAQ深度学习不会打断点啊!)。那我的
2021-11-22 10:54:45 14658
原创 深度学习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 4011
原创 ‘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 502 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 4118 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 3126 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 376 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 455
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人