Skip to content

GitLab

Commit dd911f19 authored by rusty1s's avatar rusty1s
Browse files

fix dropout

parent 724fce4b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_geometric_autoscale/utils.py
View file @ dd911f19
...@@ -59,7 +59,7 @@ def gen_masks(y: Tensor, train_per_class: int = 20, val_per_class: int = 30, ...@@ -59,7 +59,7 @@ def gen_masks(y: Tensor, train_per_class: int = 20, val_per_class: int = 30,
def dropout(adj_t: SparseTensor, p: float, training: bool = True): def dropout(adj_t: SparseTensor, p: float, training: bool = True):
if not training: if not training or p == 0.:
return adj_t return adj_t
if adj_t.storage.value() is not None: if adj_t.storage.value() is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment