Pretrained GRPF(LRR-4x, ...) models

dlsrbgg33 · Jun 13, 2018 · 9238d54 · 9238d54
1 parent d0f37b3
commit 9238d54
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/evaluate.py b/evaluate.py
@@ -135,7 +135,7 @@ def evaluate(args):
 
     parser.add_argument('--static', help='Which static network to use.', required=True)
     parser.add_argument('--flow', help='Which optical flow method to use.', required=True)
-    parser.add_argument('--frames', dest='frames', type=int, help='Number of frames to use.', default=5, required=False)
+    parser.add_argument('--frames', type=int, help='Number of frames to use.', default=5, required=False)
 
     args = parser.parse_args()
 

diff --git a/models/stgru.py b/models/stgru.py
@@ -78,7 +78,7 @@ def get_GRU_cell(self, input_image, prev_image, flow_input, h_prev, unary_input)
         return h
 
     def softmax_last_dim(self, x):
-        # apply softmax to a 4D vector along the last dimension
+        # apply softmax to a 4D tensor along the last dimension
         S = tf.shape(x)
         y = tf.reshape(x, [-1, S[4-1]])
         y = tf.nn.softmax(y)