![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 68
小王做笔记
这个作者很懒,什么都没留下…
展开
-
linux-Ubuntu系统下,tf无法使用gpu加速,如Could not load dynamic library ‘libcudart.so.10.1‘等问题
如何解决tensorflow2版本无法使用cuda-gpu加速的情况,如Could not load dynamic library 'libcudart.so.10.1' 等问题场景描述版本环境:测试脚本返回错误报告尝试解决方法:1. 重新安装 cuda 和 cuda驱动2. 不重装解决策略- step1. 找到报错提示:并且发现之前安装cuda的默认路径,进入该路径内,在界面中输入:- step 2. 对需要的文件进行复制!!!**将只有版本编号不同的报错的文件进行复制!!**- step 3. 安装以原创 2021-12-05 14:08:51 · 2378 阅读 · 0 评论 -
keras报错 : ValueError: logits and labels must have the same shape
keras报错 :ValueError: logits and labels must have the same shape问题背景问题原因解决办法整体代码问题背景在一步一步搭建 文本情感分类器的过程时,将数据处理成向量的形式;之后送入到 MLP模型中进行拟合训练;发生错误:ValueError: logits and labels must have the same shape ((None, 1) vs (None, 2))问题原因原因:logits 和 labels 需要有相同的原创 2021-09-23 15:32:14 · 11402 阅读 · 0 评论 -
模型、参数和文件的保存方法
模型、参数和文件的保存方法 numpy保存 npy 文件tf.keras保存 模型和参数h5 文件保存模型和权重存在 h.5 文件分别保存模型结构和权重,并写成 h.5 文件ckpt 文件 (推荐使用)pytorch 保存模型和参数pkl 文件保存模型参数(推荐)保存完整模型 (不推荐)numpy保存 npy 文件import numpy as npnp.save(文件名, 变量名)变量名 = np.load(文件名)更详细内容:https://blog.csdn.net/leilei740原创 2021-09-16 20:30:34 · 2411 阅读 · 0 评论