ソースを参照

nn.SiLU() citation correction (#1713)

5.0
Glenn Jocher GitHub 3年前
コミット
fea9c9b80c
この署名に対応する既知のキーがデータベースに存在しません GPGキーID: 4AEE18F83AFDEB23
1個のファイルの変更1行の追加1行の削除
  1. +1
    -1
      utils/activations.py

+ 1
- 1
utils/activations.py ファイルの表示

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

読み込み中…
キャンセル
保存