Remove unused arguments

Lightning-Universe · Dec 1, 2020 · 89f7cbc · 89f7cbc
1 parent 2e903c3
commit 89f7cbc
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/pl_bolts/models/detection/faster_rcnn.py b/pl_bolts/models/detection/faster_rcnn.py
@@ -122,9 +122,6 @@ def add_model_specific_args(parent_parser):
         parser.add_argument("--pretrained_backbone", type=bool, default=True)
         parser.add_argument("--trainable_backbone_layers", type=int, default=3)
         parser.add_argument("--replace_head", type=bool, default=True)
-
-        parser.add_argument("--data_dir", type=str, default=".")
-        parser.add_argument("--batch_size", type=int, default=1)
         return parser