Pytorch
蜗牛在听雨
这个作者很懒,什么都没留下…
展开
-
pytorhc模型权重导出 解析.npy文件 将数据按照指定格式打印到txt中
代码如下:import numpy as npinput_data = np.load(r"mlp_6_weight.npy")print(input_data.shape)data = input_data.T #根据自己的需求确认是否要转置print(data.shape)print(data)np.savetxt(r"mlp_6_weight.txt",data,fmt=['%.017f']*data.shape[1],delimiter=',',newline='},\n{')原创 2020-11-05 17:19:41 · 889 阅读 · 0 评论 -
VS2019 + Qt 下 libtorch配置报错:C2059 C2334
一、程序配置好libtorch环境后,采用如下的程序进行测试:#include <torch/torch.h>#include <iostream>int main() { torch::Tensor tensor = torch::eye(3); std::cout << tensor << std::endl;}二、报错点击两个错误的地方,ivalue_inl.h头文件内容如下:object.h 头文件内容如下:三、解决原创 2020-10-23 21:05:01 · 1357 阅读 · 7 评论 -
深度学习 解决过拟合问题
一、问题在深度学习时,一般将数据集划分成训练集和测试集。如果最终的训练结果不满意,一般就是以下两种情况:(1)训练集的准确率不高,测试集的准确率也不高;(欠拟合)(2)训练集的准确率非常高,测试集的准确率比训练集差上许多。(过拟合)对于第(1)种情况,一般是因为模型自身的问题,比如模型本身选择的不对,模型的类型不对或者模型的深度太浅了,需要重新选择合适的模型。对于第(2)种情况,通过训练过程中的损失值变化可以判断出过拟合情况的发生。例如下面的两个图,训练集上的损失一直在下降,但验证集的损失函数在某原创 2020-09-29 10:30:36 · 1211 阅读 · 0 评论 -
Anaconda 安装 torch_scatter 失败的解决方法
一、解决方法在做图卷积神经网络时会用到 torch-scatter、torch-sparse 等包,但直接用conda命令安装,会出现找不到源的问题。解决方法是直接去官网下载whl安装文件。pytorch-geometric官网地址根据要安装的包的名字、安装包版本号、cuda版本号、python版本号、操作系统和位数,找到对应的whl安装包进行下载。二、安装找到whl安装包文件的下载路径,使用pip命令进行安装pip install G:\GitHub\Feature_SAG\torch_sc原创 2020-09-25 18:38:22 · 5934 阅读 · 0 评论 -
Anaconda中安装Pytorch 速度太慢解决办法
一、问题在Anaconda中新建环境后,想在GUI上直接安装Pytorch,但是速度特别慢。二、解决方法1.点击环境右侧的箭头,打开当前环境的终端2. 添加镜像在弹出来的终端窗口中添加镜像网址,输入的代码如下:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/conda config --add channels https://mirrors.tuna.tsing原创 2020-09-17 21:22:04 · 16442 阅读 · 4 评论