Fix small typo (relu instead of ReLU).

This commit is contained in:
Austin J. Garrett
2018-04-08 20:18:43 -04:00
parent e9102f420e
commit 20af804ba0
+1 -1
View File
@@ -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).