From 9533bf0afa483d9bbe5c218e74fc55f7040ef1a9 Mon Sep 17 00:00:00 2001 From: mgqa34 Date: Sun, 25 Jun 2023 17:15:15 +0800 Subject: [PATCH] fix gpt2 guide: add trainer_param and datasetparam import, see issue #12 Signed-off-by: mgqa34 --- doc/tutorial/GPT2-example.ipynb | 2 ++ 1 file changed, 2 insertions(+) diff --git a/doc/tutorial/GPT2-example.ipynb b/doc/tutorial/GPT2-example.ipynb index 1b540ff..09f3ee7 100644 --- 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",