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 阅读(313) 评论(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 阅读(486) 评论(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 阅读(1496) 评论(0) 推荐(0) 编辑