summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--cli/app/search/search_dense.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index 2b51977..ecce837 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -26,8 +26,8 @@ feature_layer_names = {
'1a': "InceptionV3/Conv2d_1a_3x3",
'2a': "InceptionV3/Conv2d_2a_3x3",
'2b': "InceptionV3/Conv2d_2b_3x3",
- '3a': "InceptionV3/Conv2d_3b_3x3",
- '3b': "InceptionV3/Conv2d_3b_3x3",
+ '3a': "InceptionV3/Conv2d_3b_1x1",
+ '3b': "InceptionV3/Conv2d_3b_1x1",
'4a': "InceptionV3/Conv2d_4a_3x3",
'5a': "InceptionV3/Mixed_5a",
'5b': "InceptionV3/Mixed_5b",