fix(graph): correctly discard dropout

jolibrain · Nov 3, 2020 · 16409a6 · 16409a6
1 parent 55ee5c8
commit 16409a6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/generators/net_caffe_recurrent.cc b/src/generators/net_caffe_recurrent.cc
@@ -434,6 +434,7 @@ namespace dd
                      top);
             add_tile(this->_dnet_params, "tile_" + std::to_string(i), prevname,
                      top);
+            dropouts.insert(dropouts.begin() + i, 0.0);
           }
         else
           {