Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions examples/multi_gpu/pcqm4m_ogb.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,7 +117,7 @@ def __init__(self, num_layers, emb_dim, drop_ratio=0.5, JK="last",
num_layers (int): number of GNN message passing layers.
residual (bool): whether to add residual connection.
drop_ratio (float): dropout ratio.
JK (str): "last" or "sum" to choose JK concat strat.
JK (str): "last" or "sum" to choose JK concat strata.
residual (bool): Whether or not to add the residual
gnn_type (str): Type of GNN to use.
"""
Expand Down Expand Up @@ -302,8 +302,8 @@ def __init__(
virtual_node (bool): whether to add virtual node or not.
residual (bool): Whether or not to add the residual
drop_ratio (float): dropout ratio.
JK (str): "last" or "sum" to choose JK concat strat.
graph_pooling (str): Graph pooling strat to use.
JK (str): "last" or "sum" to choose JK concat strata.
graph_pooling (str): Graph pooling strata to use.
"""
super().__init__()
if num_layers < 2:
Expand Down
4 changes: 2 additions & 2 deletions examples/randlanet_classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,9 +62,9 @@ def message(self, x_j: Tensor, pos_i: Tensor, pos_j: Tensor,
"""Local Spatial Encoding (locSE) and attentive pooling of features.

Args:
x_j (Tensor): neighboors features (K,d)
x_j (Tensor): neighbors features (K,d)
pos_i (Tensor): centroid position (repeated) (K,3)
pos_j (Tensor): neighboors positions (K,3)
pos_j (Tensor): neighbors positions (K,3)
index (Tensor): index of centroid positions
(e.g. [0,...,0,1,...,1,...,N,...,N])

Expand Down
2 changes: 1 addition & 1 deletion graphgym/custom_graphgym/optimizer/example.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ def adagrad_optimizer(params: Iterator[Parameter], base_lr: float,
return Adagrad(params, lr=base_lr, weight_decay=weight_decay)


@register.register_scheduler('pleateau')
@register.register_scheduler('plateau')
def plateau_scheduler(optimizer: Optimizer, patience: int,
lr_decay: float) -> ReduceLROnPlateau:
return ReduceLROnPlateau(optimizer, patience=patience, factor=lr_decay)
2 changes: 1 addition & 1 deletion torch_geometric/data/lightning/datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,7 @@ def __init__(

if loader in ['neighbor', 'link_neighbor']:

# Define a new `NeighborSampler` to be re-used across data loaders:
# Define a new `NeighborSampler` to be reused across data loaders:
sampler_kwargs, self.loader_kwargs = split_kwargs(
self.kwargs,
NeighborSampler,
Expand Down
2 changes: 1 addition & 1 deletion torch_geometric/graphgym/models/gnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ def __init__(self, dim_in: int):
has_bias=False,
cfg=cfg,
))
# Update `dim_in` to reflect the new dimension fo the node features
# Update `dim_in` to reflect the new dimension of the node features
self.dim_in = cfg.gnn.dim_inner
if cfg.dataset.edge_encoder:
# Encode integer edge features via `torch.nn.Embedding`:
Expand Down
Loading