change LSTM_test

ilovin · Aug 8, 2017 · cc7083e · cc7083e
1 parent df0ed3d
commit cc7083e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/lib/networks/LSTM_test.py b/lib/networks/LSTM_test.py
@@ -17,9 +17,10 @@ def __init__(self, trainable=True):
 
     def setup(self):
         (self.feed('data')
-         .conv_single(5, 5, 32 ,1, 1, name='conv1',c_i=cfg.NCHANNELS)
-         .conv_single(5, 5, 32 ,1, 1, name='conv2')
+         .conv_single(3, 3, 32 ,1, 1, name='conv1',c_i=cfg.NCHANNELS)
+         .conv_single(3, 3, 64 ,1, 1, name='conv2')
          .max_pool(2, 2, 2, 2, padding='VALID', name='pool1')
-         .conv_single(5, 5, 1 ,1, 1, name='conv4',relu=False))
+         .conv_single(3, 3, 1 ,1, 1, name='conv4',relu=False))
         (self.feed('conv4','time_step_len')
-         .lstm(cfg.TRAIN.NUM_HID,cfg.TRAIN.NUM_LAYERS,name='logits',img_shape=[-1,cfg.IMG_SHAPE[0]//cfg.POOL_SCALE,cfg.NUM_FEATURES//cfg.POOL_SCALE]))
+         .bi_lstm(cfg.TRAIN.NUM_HID,cfg.TRAIN.NUM_LAYERS,name='logits',img_shape=[-1,cfg.IMG_SHAPE[0]//cfg.POOL_SCALE,cfg.NUM_FEATURES//cfg.POOL_SCALE]))
+         #.lstm(cfg.TRAIN.NUM_HID,cfg.TRAIN.NUM_LAYERS,name='logits',img_shape=[-1,cfg.IMG_SHAPE[0]//cfg.POOL_SCALE,cfg.NUM_FEATURES//cfg.POOL_SCALE]))