zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

nn.SiLU() citation correction (#1713)

This commit is contained in:
Glenn Jocher 2021-01-06 20:58:41 -08:00 committed by GitHub
parent e77c77f580
commit fea9c9b80c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/activations.py
View File

@ -5,7 +5,7 @@ import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
# SiLU https://arxiv.org/pdf/1905.02244.pdf ---------------------------------------------------------------------------- # SiLU https://arxiv.org/pdf/1606.08415.pdf ----------------------------------------------------------------------------
class SiLU(nn.Module): # export-friendly version of nn.SiLU() class SiLU(nn.Module): # export-friendly version of nn.SiLU()
@staticmethod @staticmethod
def forward(x): def forward(x):