Set default --layer arg in clf_voc07.py to avgpool.

kdexd · Mar 3, 2021 · 5b2f12c · 5b2f12c
1 parent f940d0e
commit 5b2f12c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/clf_voc07.py b/scripts/clf_voc07.py
@@ -38,7 +38,7 @@
 parser.add_argument_group("Checkpointing")
 group.add_argument(
     "--layer", choices=["layer1", "layer2", "layer3", "layer4", "avgpool"],
-    default="layer4", help="Evaluate features extracted from this layer."
+    default="avgpool", help="Evaluate features extracted from this layer."
 )
 parser.add_argument(
     "--weight-init", choices=["random", "imagenet", "torchvision", "virtex"],