diff --git a/pl_bolts/models/detection/faster_rcnn.py b/pl_bolts/models/detection/faster_rcnn.py index b6b6674bf8..e7defbb0e4 100644 --- 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