Merge pull request #4806 from FederatedAI/feature-1.11.1-graph-bug-fix

Feature 1.11.1 graph bug fix
FederatedAI · Apr 19, 2023 · 50be383 · 50be383
2 parents 391415e + 95ac058
commit 50be383
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 8 deletions.
diff --git a/doc/tutorial/pipeline/nn_tutorial/Homo-Graph.ipynb b/doc/tutorial/pipeline/nn_tutorial/Homo-Graph.ipynb
@@ -160,7 +160,7 @@
     "from pipeline.component.nn import DatasetParam\n",
     "\n",
     "fate_torch_hook(t)\n",
-    "fate_project_path = os.getcwd() + \"/../../../../\"\n",
+    "fate_project_path = os.path.abspath(\"/../../../../\")\n",
     "host = 10000\n",
     "guest = 9999\n",
     "arbiter = 10000\n",
@@ -169,8 +169,8 @@
     "data_0 = {\"name\": \"cora_guest\", \"namespace\": \"experiment\"}\n",
     "data_1 = {\"name\": \"cora_host\", \"namespace\": \"experiment\"}\n",
     "\n",
-    "data_path_0 = fate_project_path + 'examples/data/cora4fate/guest'\n",
-    "data_path_1 = fate_project_path + 'examples/data/cora4fate/host'\n",
+    "data_path_0 = fate_project_path + '/examples/data/cora4fate/guest'\n",
+    "data_path_1 = fate_project_path + '/examples/data/cora4fate/host'\n",
     " "
    ]
   },
@@ -286,7 +286,7 @@
    "hash": "6014097d4fbe68b982926f809650a6d6f12d367fa066105c39b397ad90be87cf"
   },
   "kernelspec": {
-   "display_name": "Python 3.8.8 ('venv': venv)",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },
@@ -300,9 +300,8 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.8"
-  },
-  "orig_nbformat": 4
+   "version": "3.9.13"
+  }
  },
  "nbformat": 4,
  "nbformat_minor": 2

diff --git a/doc/tutorial/pipeline/nn_tutorial/README.md b/doc/tutorial/pipeline/nn_tutorial/README.md
@@ -83,6 +83,10 @@ Here we offer some advanced examples of using FATE-NN framework.
 
 - [Using Frozen Parameters Bert for Sentiment Classification](Bert-example.ipynb)
 
+## Node Classification Using GNN(Homo-NN)
+
+- [Homo Graph Sage for Node Classification Task](Homo-Graph.ipynb)
+
 ## Training on vertical-split heterogeneous data(Hetero-NN)
 
 - [A federated task with guest using image data and host using text data](Hetero-NN-Mix-Task.ipynb)
diff --git a/python/federatedml/nn/homo/trainer/fedavg_graph_trainer.py b/python/federatedml/nn/homo/trainer/fedavg_graph_trainer.py
@@ -46,7 +46,7 @@ class FedAVGGraphTrainer(FedAVGTrainer):
     def __init__(self, epochs=10, batch_size=512,  # training parameter
                  early_stop=None, tol=0.0001,  # early stop parameters
                  secure_aggregate=True, weighted_aggregation=True, aggregate_every_n_epoch=None,  # federation
-                 cuda=False, pin_memory=True, shuffle=True, data_loader_worker=0,  # GPU & dataloader
+                 cuda=None, pin_memory=True, shuffle=True, data_loader_worker=0,  # GPU & dataloader
                  validation_freqs=None,  # validation configuration
                  checkpoint_save_freqs=None,  # checkpoint configuration
                  task_type='auto',