DynamicReLU:动态ReLU在Pytorch上的实现

时间:2024-04-08 11:48:59
【文件属性】:

文件名称:DynamicReLU:动态ReLU在Pytorch上的实现

文件大小:1KB

文件格式:ZIP

更新时间:2024-04-08 11:48:59

dynamic pytorch relu Python

动态ReLU 在Pytorch上实现 。 例子 import torch.nn as nn from dyrelu import DyReluB class Model(nn.Module): def __init__(self): super(Model, self).__init__() self.conv1 = nn.Conv2d(3, 10, 5) self.relu = DyReLUB(10, conv_type='2d') def forward(self, x): x = self.conv1(x) x = self.relu(x) return x


【文件预览】:
DynamicReLU-master
----dyrelu.py(3KB)
----README.md(444B)

网友评论