
深度学习
文章平均质量分 75
路遥_w
这个作者很懒,什么都没留下…
展开
-
VSCODE调试控制台的使用——以pytorch下神经网络的加载模型前向传播为例
最近在做模型的量化与量化后模型在arm平台上的迁移工作,为了保证量化后的模型在python下与在arm下运行的一致性(相同输入下,不同环境下的输出完全一致),需要对python下模型前向传播的每个操作的激活值、权重、输入的量化参数(scale与zero_point)、权重的量化参数(scale与zero_point)都能查看,在这个查看的过程中,发现VSCODE帮了大忙,因此记录一下VSCODE调试控制台的使用,便于自己与他人后续参考使用。原创 2022-09-22 14:37:25 · 2259 阅读 · 1 评论 -
在线难例挖掘:Online Hard Example Mining (OHEM)
详细链接:https://erogol.com/online-hard-example-mining-pytorch/OHEM通过减少计算成本来选择难例,提高网络性能。它主要用于目标检测。假设你想训练一个汽车检测器,并且你有正样本图像(图像中有汽车)和负样本图像(图像中没有汽车)。现在你想训练你的网络。实际上,你会发现负样本数量会远远大于正样本数量,而且大量的负样本是比较简单的。因此,比较明智的做法是选择一部分对网络最有帮助的负样本(比较有难度的,容易被识别为正样本的)参与训练。难例挖掘就是用于选择对网原创 2021-07-27 16:32:05 · 2101 阅读 · 0 评论 -
Pytorch网络模型权重初始化、保存与加载模型、加载预训练模型、按需设置学习率
权重初始化加载预训练模型原创 2021-01-20 18:08:39 · 10188 阅读 · 1 评论