gaspersic/gpt-2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

push 774M model

Browse Source
This commit is contained in:
Jeff Wu
2019-08-20 08:50:19 -07:00
parent cb415376c3
commit f35fa1d920
7 changed files with 20 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/generate_unconditional_samples.py
View File

@ -9,7 +9,7 @@ import tensorflow as tf
import model, sample, encoder
def sample_model(
model_name='117M',
model_name='124M',
seed=None,
nsamples=0,
batch_size=1,
@ -20,7 +20,7 @@ def sample_model(
):
"""
Run the sample_model
:model_name=117M : String, which model to use
:model_name=124M : String, which model to use
:seed=None : Integer seed for random number generators, fix seed to
reproduce results
:nsamples=0 : Number of samples to return, if 0, continues to

8
src/interactive_conditional_samples.py
View File

@ -9,18 +9,18 @@ import tensorflow as tf
import model, sample, encoder
def interact_model(
model_name='117M',
model_name='124M',
seed=None,
nsamples=1,
batch_size=1,
length=None,
temperature=1,
top_k=0,
models_dir='models',
models_dir='models',
):
"""
Interactively run the model
:model_name=117M : String, which model to use
:model_name=124M : String, which model to use
:seed=None : Integer seed for random number generators, fix seed to reproduce
results
:nsamples=1 : Number of samples to return total
@ -36,7 +36,7 @@ def interact_model(
while 40 means 40 words are considered at each step. 0 (default) is a
special setting meaning no restrictions. 40 generally is a good value.
:models_dir : path to parent folder containing model subfolders
(i.e. contains the <model_name> folder)
(i.e. contains the <model_name> folder)
"""
models_dir = os.path.expanduser(os.path.expandvars(models_dir))
if batch_size is None: