From 20af804ba0809e35feb356ec90a698e3f3327ba4 Mon Sep 17 00:00:00 2001 From: "Austin J. Garrett" Date: Sun, 8 Apr 2018 20:18:43 -0400 Subject: [PATCH] Fix small typo (relu instead of ReLU). --- core/util_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/util_layers.py b/core/util_layers.py index 290fddb..684af98 100644 --- a/core/util_layers.py +++ b/core/util_layers.py @@ -59,7 +59,7 @@ class Conv(nn.Module): def __init__(self, in_channels : int, out_channels : int, kernel_size : Union[int, Tuple[int, int]], with_bn : bool = True, - activation : Callable[[UFloatTensor], UFloatTensor] = nn.relu() + activation : Callable[[UFloatTensor], UFloatTensor] = nn.ReLU() ) -> None: """ :param in_channels: Length of input featuers (first dimension).