fix gpt2 guide: add trainer_param and datasetparam import, see issue #12

Signed-off-by: mgqa34 <mgq3374541@163.com>
FederatedAI · Jun 25, 2023 · 9533bf0 · 9533bf0
1 parent 10e95f8
commit 9533bf0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/doc/tutorial/GPT2-example.ipynb b/doc/tutorial/GPT2-example.ipynb
@@ -521,11 +521,13 @@
     "import os\n",
     "from pipeline import fate_torch_hook\n",
     "from pipeline.component import HomoNN\n",
+    "from pipeline.component.homo_nn import DatasetParam, TrainerParam\n",
     "from pipeline.backend.pipeline import PipeLine\n",
     "from pipeline.component import Reader\n",
     "from pipeline.interface import Data\n",
     "from transformers import GPT2Config\n",
     "\n",
+    "\n",
     "fate_torch_hook(t)\n",
     "\n",
     "\n",