From ff9a21d333f11a42e67c1963ed67de9c0fda87c9 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Thu, 7 Jan 2021 20:10:54 +0100
Subject: Minor updates.

---
 src/text_recognizer/networks/util.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'src/text_recognizer/networks/util.py')

diff --git a/src/text_recognizer/networks/util.py b/src/text_recognizer/networks/util.py
index e2d7955..711a952 100644
--- a/src/text_recognizer/networks/util.py
+++ b/src/text_recognizer/networks/util.py
@@ -39,6 +39,7 @@ def activation_function(activation: str) -> Type[nn.Module]:
         [
             ["elu", nn.ELU(inplace=True)],
             ["gelu", nn.GELU()],
+            ["glu", nn.GLU()],
             ["leaky_relu", nn.LeakyReLU(negative_slope=1.0e-2, inplace=True)],
             ["none", nn.Identity()],
             ["relu", nn.ReLU(inplace=True)],
-- 
cgit v1.2.3-70-g09d2