Improved docstrings in model module

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-02-08 22:42:23 +01:00
parent 1e422d3fa7
commit 9909b5fdfc

View File

@ -64,16 +64,7 @@ class Encoder(keras.Model):
padding='valid', kernel_initializer=weight_init) padding='valid', kernel_initializer=weight_init)
def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor: def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor:
""" """See base class."""
Overwrites the ``call`` method and is called by ``__call__``.
Args:
inputs: input values
``**kwargs``: additional keyword arguments - none are used
Returns:
result values
"""
result = self.x_padded(inputs) result = self.x_padded(inputs)
result = self.conv1(result) result = self.conv1(result)
result = self.conv1_a(result) result = self.conv1_a(result)
@ -120,16 +111,7 @@ class Decoder(keras.Model):
self.deconv4_cropped = keras.layers.Cropping2D(cropping=1) self.deconv4_cropped = keras.layers.Cropping2D(cropping=1)
def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor: def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor:
""" """See base class."""
Overwrites the ``call`` method and is called by ``__call__``.
Args:
inputs: input values
``**kwargs``: additional keyword arguments - none are used
Returns:
result values
"""
result = self.deconv1(inputs) result = self.deconv1(inputs)
result = self.deconv1_bn(result) result = self.deconv1_bn(result)
result = self.deconv1_a(result) result = self.deconv1_a(result)
@ -167,16 +149,7 @@ class ZDiscriminator(keras.Model):
kernel_initializer=weight_init) kernel_initializer=weight_init)
def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor: def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor:
""" """See base class."""
Overwrites the ``call`` method and is called by ``__call__``.
Args:
inputs: input values
``**kwargs``: additional keyword arguments - none are used
Returns:
result values
"""
result = self.zd1(inputs) result = self.zd1(inputs)
result = self.zd1_a(result) result = self.zd1_a(result)
result = self.zd2(result) result = self.zd2(result)
@ -213,16 +186,7 @@ class XDiscriminator(keras.Model):
activation='sigmoid') activation='sigmoid')
def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor: def call(self, inputs: tf.Tensor, **kwargs) -> tf.Tensor:
""" """See base class."""
Overwrites the ``call`` method and is called by ``__call__``.
Args:
inputs: input values
``**kwargs``: additional keyword arguments - none are used
Returns:
result values
"""
result = self.x_padded(inputs) result = self.x_padded(inputs)
result = self.xd1(result) result = self.xd1(result)
result = self.xd1_a(result) result = self.xd1_a(result)