0
点赞
收藏
分享

微信扫一扫

【PyTorch】SiLU激活函数


问题

方法

SiLU激活函数介绍

【PyTorch】SiLU激活函数_python

SiLU激活函数介绍

import torch
from torch import nn

class Net(nn.Module):

def __init__(self) -> None:
super().__init__()

self.conv = nn.Conv2d(3, 32, 3, padding=1, stride=1)
self.silu = nn.SiLU()

def forward(self, x):

x = self.conv(x)
out = self.silu(x)

return out

if __name__ == '__main__':

import netron

device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
x = torch.randn(1, 3, 224, 224).to(device)

net = Net()

model_file = 'demo.pth'
torch.onnx.export(net, x, model_file)
netron.start(model_file)

结语


举报

相关推荐

0 条评论