diff options
| -rw-r--r-- | Code/d_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/Code/d_model.py b/Code/d_model.py index 7b1cb12..1345ceb 100644 --- a/Code/d_model.py +++ b/Code/d_model.py @@ -12,7 +12,7 @@ class DiscriminatorModel: def __init__(self, session, summary_writer, height, width, scale_conv_layer_fms, scale_kernel_sizes, scale_fc_layer_sizes): """ - Initializes a GeneratorModel. + Initializes a DiscriminatorModel. @param session: The TensorFlow session. @param summary_writer: The writer object to record TensorBoard summaries @@ -22,7 +22,7 @@ class DiscriminatorModel: scale network. @param scale_kernel_sizes: The size of the kernel for each layer of each scale network. @param scale_fc_layer_sizes: The number of nodes in each fully-connected layer of each scale - network. + network. @type session: tf.Session @type summary_writer: tf.train.SummaryWriter |
