SisterRu

导航

 

2021年3月17日

摘要: 1、ravel()函数和flatten()函数 二者功能是一致的,将多维数组降为一维。区别在于flatten()函数返回的是拷贝,对拷贝所做的修改不会影响原始矩阵,而ravel()函数返回的是视图,类似于引用,会影响原始矩阵。 阅读全文
posted @ 2021-03-17 21:06 yiruzhao 阅读(66) 评论(0) 推荐(0) 编辑
 
摘要: 一、参数量 参数量和特征图的尺寸无关,仅和卷积核的大小,偏置及BN层有关,对于卷积张量kernel=(K, S, C_in, C_out),权重参数量为 params = C_out * (K * K * C_in + 1)。 阅读全文
posted @ 2021-03-17 17:02 yiruzhao 阅读(311) 评论(0) 推荐(0) 编辑
 
摘要: 转载自https://blog.csdn.net/qq_31347869/article/details/100566719 torchvision 是 pytorch 中一个很好用的包,主要由 3 个子包,分别是 torchvision.datasets,torchvision.models 和 阅读全文
posted @ 2021-03-17 16:56 yiruzhao 阅读(477) 评论(0) 推荐(0) 编辑
 
摘要: 转载自https://mp.weixin.qq.com/s?__biz=MzIzNjc1NzUzMw==&mid=2247485927&idx=1&sn=606f87b569a0eb7c29f1c2374ff698dd&chksm=e8d3be95dfa43783ddbd577ce8cc7db64c 阅读全文
posted @ 2021-03-17 10:38 yiruzhao 阅读(42) 评论(0) 推荐(0) 编辑
 
摘要: 在神经网络中,参数默认是进行随机初始化的。如果不设置的话每次训练时的初始化都是随机的,导致结果不确定。如果设置初始化,则每次初始化都是固定的。 def same_seeds(seed): torch.manual_seed(seed) if torch.cuda.is_available(): to 阅读全文
posted @ 2021-03-17 08:25 yiruzhao 阅读(1488) 评论(0) 推荐(0) 编辑
 

2021年3月16日

摘要: Top-1 error 的意思是:假如模型预测某张动物图片(一只猫)的类别,且模型只输出1个预测结果,那么这一个结果正好能猜出来这个动物是只猫的概率就是Top-1正确率。 猜出来的结果不是猫的概率则成为Top-1错误率。简单来说就是模型猜错的概率。 Top-5 error 的意思是:假如模型预测某张 阅读全文
posted @ 2021-03-16 15:56 yiruzhao 阅读(1437) 评论(0) 推荐(0) 编辑
 
摘要: 1、保存文件的后缀 有.pt,.pth,.pkl,在格式上没有任何区别,都能用,随便哪种都行。 2、保存和加载整个模型 torch.save(net, 'net_model.pkl') 加载模型 net_loaded = torch.load('net_model.pkl') 3、只保存模型参数 t 阅读全文
posted @ 2021-03-16 14:31 yiruzhao 阅读(115) 评论(0) 推荐(0) 编辑
 

2021年3月15日

摘要: 1、nn.Conv2d class torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True) 二维卷积层, 输入的尺度是(N, C_in, 阅读全文
posted @ 2021-03-15 20:38 yiruzhao 阅读(1297) 评论(0) 推荐(0) 编辑
 
摘要: 1、squeeze()函数和unsqueeze()函数 首先要知道tensor的维度,比如tensor([[0, 1, 2],[ 3, 4, 5]]),维度是(2, 3),当其维度变为(2, 1, 3)时,代表2个1行3列的矩阵,为tensor([[[0, 1, 2],[[3, 4, 5]]])。 阅读全文
posted @ 2021-03-15 20:09 yiruzhao 阅读(107) 评论(0) 推荐(0) 编辑
 

2020年12月12日

摘要: 一、下载安装 下载安装有两种方法,一种是在mac的App Store,但是中国大陆这边不支持,没有这个App;第二种是通过下载Beta版,HockeyApp 网页端下载,从该处下载的软件,属于测试(beta)版本,里面包含一些最新的特性。我还用百度网盘下载过8.x的,但那是旧版的,不能用。 二、配置 阅读全文
posted @ 2020-12-12 16:26 yiruzhao 阅读(1727) 评论(0) 推荐(0) 编辑