add docstring for network

furushchev · Jul 17, 2018 · 336dd90 · 336dd90
1 parent 2810be1
commit 336dd90
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/chainervr/models/conv_lstm/conv_lstm.py b/chainervr/models/conv_lstm/conv_lstm.py
@@ -12,6 +12,7 @@
 
 
 class ConvLSTM(chainer.Chain):
+    """Reconstruction-Prediction Network using Convolutional LSTM"""
     def __init__(self, n_channels, hidden_channels=None,
                  in_episodes=None, out_episodes=None,
                  patch_size=(128, 128),

diff --git a/chainervr/models/unsupervised_videos/lstm.py b/chainervr/models/unsupervised_videos/lstm.py
@@ -9,6 +9,7 @@
 
 
 class RPLSTM(chainer.Chain):
+    """Reconstruction-Prediction LSTM"""
     def __init__(self, n_channels, patch_size, n_layers=None, n_hidden=None, predict=True,
                  in_episodes=None, out_episodes=None):
         super(RPLSTM, self).__init__()