From fcaccc8730086135246b2236033145cc21eaed46 Mon Sep 17 00:00:00 2001 From: Stanislaw Adaszewski Date: Fri, 24 Jul 2020 13:28:11 +0200 Subject: [PATCH] Fix a typo. --- src/icosagon/fastconv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/icosagon/fastconv.py b/src/icosagon/fastconv.py index 02086c4..6f90eb8 100644 --- a/src/icosagon/fastconv.py +++ b/src/icosagon/fastconv.py @@ -81,7 +81,7 @@ class FastGraphConv(torch.nn.Module): in_channels: List[int], out_channels: List[int], data: Union[Data, PreparedData], - relation_family: Union[RelationFamily, PreparedRelationFamily] + relation_family: Union[RelationFamily, PreparedRelationFamily], keep_prob: float = 1., acivation: Callable[[torch.Tensor], torch.Tensor] = lambda x: x, **kwargs) -> None: