enkaell commited on
Commit
4a55c52
·
1 Parent(s): e742ed1

Delete fine-tuning.py

Browse files
Files changed (1) hide show
  1. fine-tuning.py +0 -31
fine-tuning.py DELETED
@@ -1,31 +0,0 @@
1
- import gpt_2_simple as gpt2
2
- import configparser
3
- config = configparser.ConfigParser()
4
- config.read("conf.ini")
5
-
6
-
7
- file_name = "data.txt"
8
- run_name = config["DEFAULT"]["RunName"]
9
- model_size = config["DEFAULT"]["GPT2ModelNameSize"]
10
- steps = config["DEFAULT"]["Steps"]
11
-
12
- gpt2.download_gpt2(model_name=model_size)
13
- sess = gpt2.start_tf_sess()
14
-
15
- gpt2.finetune(sess,
16
- dataset=file_name,
17
- model_name=model_size,
18
- steps=steps,
19
- restore_from='fresh',
20
- run_name = run_name,
21
- print_every=10,
22
- )
23
- # , learning_rate=.00003)
24
-
25
-
26
- print(run_name)
27
- sess = gpt2.start_tf_sess()
28
- gpt2.load_gpt2(sess, run_name=run_name)
29
-
30
-
31
- gpt2.generate(sess, run_name=run_name, temperature=0.7, length=100, prefix=None, top_k=40, nsamples=10)