Pytorch
文章平均质量分 86
Midsummer-逐梦
这个作者很懒,什么都没留下…
展开
-
yolov9从头开始训练
本数量,如果你GPU性能可以上天,显存可以下地那么可以调大一下,目前我设置为64,输入单样本大小为480×640480×640,消耗显存45G;如果显存小就调小点。时过境迁,当读者你看到此教程时,可能距离作者收笔此文相去已久,若未能解决你的问题,作者深感抱歉。若读者发现某些地方已经不再适合当下,可以私聊或者评论区告知,我会及时进行修改!原创 2024-02-29 11:40:20 · 1400 阅读 · 0 评论 -
Pytorch学习torch.clamp ()用法浅析
input: 输入张量,即需要进行元素限制的张量。min: 张量中的元素的最小值。如果元素小于这个值,将被替换为这个最小值。max: 张量中的元素的最大值。如果元素大于这个值,将被替换为这个最大值。out(可选): 输出张量,用于保存结果。如果没有提供,函数会创建一个新的张量来保存结果。原创 2024-01-05 21:14:07 · 843 阅读 · 0 评论 -
全连接层(线性连接层)-torch.nn.Linear-学习
我写这篇文章的原因是,我之前对权重矩阵的形状在Linear层的表示不是很理解。对于上面的例子,输入形状为【3x4】,在我的理解下,权重矩阵的形状应该为【4x1】因为这样刚好就可以直接进行矩阵乘积了,很方便。我不明白之处在于为什么Linear的权重矩阵形状要设置为【1,4】,然后运算的时候先做一个转置变成【4x1】,再进行【3x4】【4x1】的矩阵乘积操作。大家有想法的在评论区留言吧,我在网上暂时没找到解释得过去的答案!原创 2023-12-31 03:11:17 · 1129 阅读 · 0 评论