wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-12 12:04:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Nick Walton
2019-11-18 18:29:38 -07:00
parent 8a049cef10
commit 55da36426a
3 changed files with 61 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/gpt2/gpt2_generator.py
+2 -2
View File
@@ -11,13 +11,13 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
class GPT2Generator:
def __init__(self, generate_num=80, temperature=0.3, top_k=40, top_p=0.8):
def __init__(self, generate_num=80, temperature=0.4, top_k=40, top_p=0.8):
self.generate_num=generate_num
self.temp = temperature
self.top_k = top_k
self.top_p = top_p
self.model_name = "model_v2"
self.model_name = "model_v1"
self.model_dir = "generator/gpt2/models"
self.checkpoint_path = os.path.join(self.model_dir, self.model_name)