diff --git a/src/LayerInfo.cpp b/src/LayerInfo.cpp index 8741053..63a2844 100644 --- a/src/LayerInfo.cpp +++ b/src/LayerInfo.cpp @@ -9,7 +9,7 @@ const unordered_map LayerInfo::NAME_TYPE_MAP = { {"ReLU", Type::ReLU}, {"Pooling", Type::Pooling}, {"InnerProduct", Type::InnerProduct}, - {"DropOut", Type::DropOut}, + {"Dropout", Type::DropOut}, {"LRN", Type::LRN}, {"Split", Type::Split}, {"Softmax", Type::Softmax}