Update Base.py

ORNL · Dec 12, 2024 · d579020 · d579020
1 parent 70fb88f
commit d579020
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hydragnn/models/Base.py b/hydragnn/models/Base.py
@@ -228,7 +228,7 @@ def _embedding(self, data):
         if self.use_global_attn:
             # encode node positional embeddings
             x = self.pos_emb(data.pe)
-            # if node features are available, generate mebeddings, concatenate with positional embeddings and map to hidden dim
+            # if node features are available, generate embeddings, concatenate with positional embeddings and map to hidden dim
             if self.input_dim:
                 x = torch.cat((self.node_emb(data.x.float()), x), 1)
                 x = self.node_lin(x)