From 9d1e704fec1726d5e051e98336b655ae8ce49242 Mon Sep 17 00:00:00 2001 From: Jeff Wu Date: Tue, 26 Feb 2019 23:05:11 -0800 Subject: [PATCH] slight fix to batch size description --- src/generate_unconditional_samples.py | 3 +-- src/interactive_conditional_samples.py | 5 ++--- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/generate_unconditional_samples.py b/src/generate_unconditional_samples.py index 2199f2b..87e2129 100755 --- a/src/generate_unconditional_samples.py +++ b/src/generate_unconditional_samples.py @@ -24,8 +24,7 @@ def sample_model( reproduce results :nsamples=0 : Number of samples to return, if 0, continues to generate samples indefinately. - :batch_size=1 : Number of batches, model runs nsamples//batch_size - times, each batch run is independent of previous run. + :batch_size=1 : Number of batches (only affects speed/memory). :length=None : Number of tokens in generated text, if None (default), is determined by model hyperparameters :temperature=1 : Float value controlling randomness in boltzmann diff --git a/src/interactive_conditional_samples.py b/src/interactive_conditional_samples.py index 74b322e..166171a 100755 --- a/src/interactive_conditional_samples.py +++ b/src/interactive_conditional_samples.py @@ -22,9 +22,8 @@ def interact_model( :model_name=117M : 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 - :batch_size=1 : Number of batches, model runs nsamples//batch_size - times, each batch run is independent of previous run. + :nsamples=1 : Number of samples to return total + :batch_size=1 : Number of batches (only affects speed/memory). Must divide nsamples. :length=None : Number of tokens in generated text, if None (default), is determined by model hyperparameters :temperature=1 : Float value controlling randomness in boltzmann