summaryrefslogtreecommitdiff
path: root/Code/d_model.py
diff options
context:
space:
mode:
Diffstat (limited to 'Code/d_model.py')
-rw-r--r--Code/d_model.py4
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