pytorch
ynxdb2002
这个作者很懒,什么都没留下…
展开
-
PreAct ResNet
class PreActBlock(nn.Module):‘’‘Pre-activation version of the BasicBlock.’’’expansion = 1def __init__(self, in_planes, planes, stride=1): super(PreActBlock, self).__init__() self.bn1 = nn.BatchNorm2d(in_planes) self.conv1 = nn.Conv2d(in_plan原创 2021-03-06 23:06:54 · 848 阅读 · 0 评论 -
pytorch yolov5 focus层
yolov5中focus的定义如下:从图像上能够更加直观地理解原创 2021-02-22 23:51:17 · 4488 阅读 · 3 评论 -
pytorch hardswish激活函数
Hardswish激活函数,在MobileNetV3架构中被提出,相较于swish函数,具有数值稳定性好,计算速度快等优点,具体原理请参考:https://arxiv.org/pdf/1905.02244.pdf原创 2021-02-22 23:33:32 · 5608 阅读 · 0 评论 -
yolov5踩坑系列-1(error:‘Hardswish‘ object has no attribute ‘inplace‘)
yolov5项目github地址:https://github.com/ultralytics/yolov5/在跑test.py的时候报错:torch.nn.modules.module.ModuleAttributeError: 'Hardswish' object has no attribute 'inplace'解决方案:将上图中的代码去掉self.inplace即可...原创 2020-11-14 22:32:54 · 7042 阅读 · 9 评论