Parcourir la source

nn.SiLU() citation correction (#1713)

5.0
Glenn Jocher GitHub il y a 3 ans
Parent
révision
fea9c9b80c
Aucune clé connue n'a été trouvée dans la base pour cette signature ID de la clé GPG: 4AEE18F83AFDEB23
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. +1
    -1
      utils/activations.py

+ 1
- 1
utils/activations.py Voir le fichier

@@ -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):

Chargement…
Annuler
Enregistrer