Delete fine-tuning.py
Browse files- 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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|