Major updates to GNN example (#2109)

* Update gnn code to comply with latest APIs * Update gnn code to cover finance dataset * typo correction * Update to use job template
NVIDIA · Oct 31, 2023 · 08a0d3f · 08a0d3f
1 parent 674fbf5
commit 08a0d3f
Show file tree

Hide file tree

Showing 32 changed files with 1,479 additions and 344 deletions.
diff --git a/examples/advanced/gnn-pt/README.md b/examples/advanced/gnn-pt/README.md
diff --git a/examples/advanced/gnn-pt/jobs/gnn-pt/app/config/config_fed_client.json b/examples/advanced/gnn-pt/jobs/gnn-pt/app/config/config_fed_client.json
diff --git a/examples/advanced/gnn-pt/jobs/gnn-pt/app/config/config_fed_server.json b/examples/advanced/gnn-pt/jobs/gnn-pt/app/config/config_fed_server.json
diff --git a/examples/advanced/gnn-pt/jobs/gnn-pt/app/custom/graphsage.py b/examples/advanced/gnn-pt/jobs/gnn-pt/app/custom/graphsage.py
diff --git a/examples/advanced/gnn-pt/jobs/gnn-pt/meta.json b/examples/advanced/gnn-pt/jobs/gnn-pt/meta.json
diff --git a/examples/advanced/gnn-pt/loss_train_fl.svg b/examples/advanced/gnn-pt/loss_train_fl.svg
diff --git a/examples/advanced/gnn-pt/loss_train_graphsage.svg b/examples/advanced/gnn-pt/loss_train_graphsage.svg
diff --git a/examples/advanced/gnn-pt/requirements.txt b/examples/advanced/gnn-pt/requirements.txt