Skip to content

Commit 16409a6

Browse files
fantessileht
authored andcommitted
fix(graph): correctly discard dropout
1 parent 55ee5c8 commit 16409a6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/generators/net_caffe_recurrent.cc

+1
Original file line numberDiff line numberDiff line change
@@ -434,6 +434,7 @@ namespace dd
434434
top);
435435
add_tile(this->_dnet_params, "tile_" + std::to_string(i), prevname,
436436
top);
437+
dropouts.insert(dropouts.begin() + i, 0.0);
437438
}
438439
else
439440
{

0 commit comments

Comments
 (0)