From f44b0248d20472b5c3db225440fe4dfd80623204 Mon Sep 17 00:00:00 2001 From: rusty1s Date: Tue, 24 May 2022 08:11:23 +0200 Subject: [PATCH] update --- torch_geometric/nn/models/dimenet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torch_geometric/nn/models/dimenet.py b/torch_geometric/nn/models/dimenet.py index cb91918a4396..4e3f44169b04 100644 --- a/torch_geometric/nn/models/dimenet.py +++ b/torch_geometric/nn/models/dimenet.py @@ -402,7 +402,7 @@ class DimeNet(torch.nn.Module): interaction blocks after the skip connection. (default: :obj:`2`) num_output_layers (int, optional): Number of linear layers for the output blocks. (default: :obj:`3`) - act (string or Callable, optional): The activation function. + act (str or Callable, optional): The activation function. (default: :obj:`"swish"`) """ @@ -634,7 +634,7 @@ class DimeNetPlusPlus(DimeNet): interaction blocks after the skip connection. (default: :obj:`2`) num_output_layers: (int, optional): Number of linear layers for the output blocks. (default: :obj:`3`) - act: (string or Callable, optional): The activation funtion. + act: (str or Callable, optional): The activation funtion. (default: :obj:`"swish"`) """