Skip to content

Commit

Permalink
docs fixed for transform layer
Browse files Browse the repository at this point in the history
  • Loading branch information
ragav committed Feb 3, 2017
1 parent 3dc6f35 commit a802411
Showing 1 changed file with 5 additions and 8 deletions.
13 changes: 5 additions & 8 deletions yann/layers/transform.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ class rotate_layer (layer):
Args:
input: An input ``theano.tensor`` variable. Even ``theano.shared`` will work as long as they
are in the following shape ``mini_batch_size, height, width, channels``
are in the following shape ``mini_batch_size, channels, height, width``
verbose: similar to the rest of the toolbox.
input_shape: ``(mini_batch_size, input_size)``
angle: value from [0,1]
Expand Down Expand Up @@ -209,20 +209,17 @@ class dropout_rotate_layer (rotate_layer):
Args:
input: An input ``theano.tensor`` variable. Even ``theano.shared`` will work as long as they
are in the following shape ``mini_batch_size, height, width, channels``
are in the following shape ``mini_batch_size, channels, height, width``
verbose: similar to the rest of the toolbox.
num_neurons: number of neurons in the layer
input_shape: ``(mini_batch_size, input_size)``
batch_norm: If provided will be used, default is ``False``.
angle: value from [0,1]
borrow: ``theano`` borrow, typically ``True``.
rng: typically ``numpy.random``.
borrow: ``theano`` borrow, typicall ``True``.
dropout_rate: ``0.5`` is the default.
Notes:
Use ``dropout_rotate_layer.output`` and ``dropout_rotate_layer.output_shape`` from
this class. ``L1`` and ``L2`` are also public and can also can be used for regularization.
The class also has in public ``w``, ``b`` and ``alpha``
which are also a list in ``params``, another property of this class.
this class.
"""
def __init__ (self,
input,
Expand Down

0 comments on commit a802411

Please sign in to comment.