|
46 | 46 | "source": [
|
47 | 47 | "#@title #模型框架加载模块\n",
|
48 | 48 | "%cd /home\n",
|
49 |
| - "!git clone -b master https://github.com/EssayKillerBrain/EssayKiller.git \n", |
| 49 | + "!git clone -b master https://github.com/EssayKillerBrain/EssayKiller_V2.git \n", |
50 | 50 | "\n",
|
51 | 51 | "#%tensorflow_version 1.5.2-GPU/TPU\n",
|
52 |
| - "!mkdir -p /home/EssayKiller/AutoWritter/finetune/trained_models\n", |
| 52 | + "!mkdir -p /home/EssayKiller_V2/LanguageNetwork/GPT2/finetune/trained_models\n", |
53 | 53 | "\n",
|
54 |
| - "%cd /home/EssayKiller/AutoWritter/finetune/\n", |
55 |
| - "!perl /home/EssayKiller/AutoWritter/scripts/gdown.pl https://drive.google.com/open?id=1ujWYTOvRLGJX0raH-f-lPZa3-RN58ZQx trained_models/model.ckpt-280000.data-00000-of-00001\n", |
56 |
| - "!wget -q --show-progress https://github.com/EssayKillerBrain/EssayKiller/releases/download/v1.0/model.ckpt-280000.index -P /home/EssayKiller/AutoWritter/finetune/models/mega\n", |
57 |
| - "!wget -q --show-progress https://github.com/EssayKillerBrain/EssayKiller/releases/download/v1.0/model.ckpt-280000.meta -P /home/EssayKiller/AutoWritter/finetune/models/mega\n", |
| 54 | + "%cd /home/EssayKiller_V2/LanguageNetwork/GPT2/finetune/\n", |
| 55 | + "!perl /home/EssayKiller_V2/LanguageNetwork/GPT2/scripts/gdown.pl https://drive.google.com/open?id=1ujWYTOvRLGJX0raH-f-lPZa3-RN58ZQx trained_models/model.ckpt-280000.data-00000-of-00001\n", |
| 56 | + "!wget -q --show-progress https://github.com/EssayKillerBrain/EssayKiller/releases/download/v1.0/model.ckpt-280000.index -P /home/EssayKiller_V2/LanguageNetwork/GPT2/finetune/models/mega\n", |
| 57 | + "!wget -q --show-progress https://github.com/EssayKillerBrain/EssayKiller/releases/download/v1.0/model.ckpt-280000.meta -P /home/EssayKiller_V2/LanguageNetwork/GPT2/finetune/models/mega\n", |
58 | 58 | "\n",
|
59 | 59 | "!echo '模型下载完成,Git项目已构建,请继续点击下方的▶️'\n",
|
60 | 60 | "# If gdown.pl failed, please uncomment following code & exec\n",
|
|
86 | 86 | "#@title #文章生成模块\n",
|
87 | 87 | "作文最小篇幅 = 1024 #@param {type:\"number\", min:800, max:1024, step:1}\n",
|
88 | 88 | "生成作文篇数 = 1 #@param {type:\"number\", min:1, max:100, step:1}\n",
|
89 |
| - "%mv /home/EssayKiller/AutoWritter/finetune/models/mega/* /home/EssayKiller/AutoWritter/finetune/trained_models/\n", |
90 |
| - "%cd /home/EssayKiller/AutoWritter/\n", |
| 89 | + "%mv /home/EssayKiller_V2/LanguageNetwork/GPT2/finetune/models/mega/* /home/EssayKiller_V2/LanguageNetwork/GPT2/finetune/trained_models/\n", |
| 90 | + "%cd /home/EssayKiller_V2/LanguageNetwork/GPT2/\n", |
91 | 91 | "!export TF_CPP_MIN_LOG_LEVEL=2\n",
|
92 | 92 | "!echo '模型加载中,请稍后......'\n",
|
93 | 93 | "!PYTHONPATH=$(pwd) python scripts/demo.py -ckpt_fn finetune/trained_models/model.ckpt-280000 -min_len $作文最小篇幅 -samples $生成作文篇数\n",
|
94 |
| - "!PYTHONPATH=$(pwd) python /home/EssayKiller/AutoFormatter/formatter.py -org_text result.txt" |
| 94 | + "!PYTHONPATH=$(pwd) python scripts/formatter.py -org_text result.txt" |
95 | 95 | ]
|
96 | 96 | }
|
97 | 97 | ],
|
|
0 commit comments