Procházet zdrojové kódy

nn.SiLU() citation correction (#1713)

5.0
Glenn Jocher GitHub před 3 roky
rodič
revize
fea9c9b80c
V databázi nebyl nalezen žádný známý klíč pro tento podpis ID GPG klíče: 4AEE18F83AFDEB23
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. +1
    -1
      utils/activations.py

+ 1
- 1
utils/activations.py Zobrazit soubor

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

Načítá se…
Zrušit
Uložit