[IE][nGraph]: Fix - Accuracy deviation on deeplabv3 (#4294)
This commit is contained in:
1 parent
45c8898321
commit
deaf1f9d93
1 file changed
+1
-4
@@ -1436,12 +1436,9 @@ InferenceEngine::details::CNNLayerCreator::CNNLayerCreator(const std::shared_ptr
|
||||
THROW_IE_EXCEPTION << "Interp do not support mode '" << interp_attrs.mode << "'";
|
||||
}
|
||||
|
||||
bool align_corners;
|
||||
auto res = std::make_shared<InferenceEngine::CNNLayer>(attrs);
|
||||
res->params = params;
|
||||
|
||||
std::istringstream(params.at("align_corners")) >> align_corners;
|
||||
res->params["align_corners"] = align_corners ? "1" : "0";
|
||||
res->params["align_corners"] = interp_attrs.align_corners ? "1" : "0";
|
||||
return res;
|
||||
});
|
||||
|
||||
|
||||
Reference in new issue
Block a user