文件名称: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)