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

slight fix to batch size description

Browse Source
This commit is contained in:
Jeff Wu
2019-02-26 23:05:11 -08:00
parent ed49f03778
commit 9d1e704fec
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/generate_unconditional_samples.py
View File

@ -24,8 +24,7 @@ def sample_model(
reproduce results reproduce results
:nsamples=0 : Number of samples to return, if 0, continues to :nsamples=0 : Number of samples to return, if 0, continues to
generate samples indefinately. generate samples indefinately.
:batch_size=1 : Number of batches, model runs nsamples//batch_size :batch_size=1 : Number of batches (only affects speed/memory).
times, each batch run is independent of previous run.
:length=None : Number of tokens in generated text, if None (default), is :length=None : Number of tokens in generated text, if None (default), is
determined by model hyperparameters determined by model hyperparameters
:temperature=1 : Float value controlling randomness in boltzmann :temperature=1 : Float value controlling randomness in boltzmann

5
src/interactive_conditional_samples.py
View File

@ -22,9 +22,8 @@ def interact_model(
:model_name=117M : String, which model to use :model_name=117M : String, which model to use
:seed=None : Integer seed for random number generators, fix seed to reproduce :seed=None : Integer seed for random number generators, fix seed to reproduce
results results
:nsamples=1 : Number of samples to return :nsamples=1 : Number of samples to return total
:batch_size=1 : Number of batches, model runs nsamples//batch_size :batch_size=1 : Number of batches (only affects speed/memory). Must divide nsamples.
times, each batch run is independent of previous run.
:length=None : Number of tokens in generated text, if None (default), is :length=None : Number of tokens in generated text, if None (default), is
determined by model hyperparameters determined by model hyperparameters
:temperature=1 : Float value controlling randomness in boltzmann :temperature=1 : Float value controlling randomness in boltzmann