上传者: 42117485
|
上传时间: 2021-10-22 18:59:15
|
文件大小: 1KB
|
文件类型: -
动态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