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

View File

@ -5,7 +5,7 @@ import torch.nn as nn
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()
@staticmethod
def forward(x):