Skip to content

Commit

Permalink
docstring fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
Aske-Rosted committed Feb 16, 2024
1 parent 8696992 commit a9aeab9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/graphnet/models/gnn/dynedge_kaggle_tito.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ def __init__(
use_global_features: bool = True,
use_post_processing_layers: bool = True,
post_processing_layer_sizes: List[int] = None,
readout_layer_sizes: List[int] = None,
readout_layer_sizes: Optional[List[int]] = None,
n_head: int = 8,
nb_neighbours: int = 8,
):
Expand All @@ -58,8 +58,8 @@ def __init__(
Options are: "min", "max", "mean", and "sum".
use_global_features: Whether to use global features after pooling.
use_post_processing_layers: Whether to use post-processing layers after the `DynTrans` layers.
post_processing_layer_sizes: (Optional) The layer sizes used in the post-processing layers. Defaults to [336, 256].
readout_layer_sizes: (Optional) The layer sizes used in the readout layers. Defaults to [256, 128].
post_processing_layer_sizes: The layer sizes used in the post-processing layers. Defaults to [336, 256].
readout_layer_sizes: The layer sizes used in the readout layers. Defaults to [256, 128].
n_head: The number of heads to use in the `DynTrans` layer.
nb_neighbours: The number of neighbours to use in the `DynTrans`
layer.
Expand Down

0 comments on commit a9aeab9

Please sign in to comment.