Merge pull request #13 from dagardner-nv/david-fsi_dgl-patch10

Fix embedding_size
nv-morpheus · Aug 29, 2023 · 062481e · 062481e
2 parents 020d2b5 + 843ed38
commit 062481e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/gnn_fraud_detection_pipeline/stages/model.py b/examples/gnn_fraud_detection_pipeline/stages/model.py
@@ -254,7 +254,7 @@ def __init__(self,
 
         # input size
         in_sizes = {
-            rel: in_size if src_type == self._target else self._embedding_size
+            rel: in_size if src_type == self._target else embedding_size
             for src_type,
             rel,
             _ in input_graph.canonical_etypes