set default pre-trained model to empty

ppwwyyxx · facebook-github-bot · commit 5b2985007a82 · 2021-09-20T12:29:27.000-07:00
Summary: Empty is a better default

Reviewed By: zhanghang1989

Differential Revision: D30980361

fbshipit-source-id: 9ad2bde6b9d9462ccf7cf953c33894a0c0279ad1
diff --git a/configs/COCO-Detection/retinanet_R_50_FPN_1x.py b/configs/COCO-Detection/retinanet_R_50_FPN_1x.py
@@ -7,3 +7,5 @@
 dataloader.train.mapper.use_instance_mask = False
 model.backbone.bottom_up.freeze_at = 2
 optimizer.lr = 0.01
+
+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
diff --git a/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x.py b/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x.py
@@ -5,3 +5,4 @@
 from ..common.models.mask_rcnn_c4 import model
 
 model.backbone.freeze_at = 2
+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
diff --git a/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.py b/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.py
@@ -5,3 +5,4 @@
 from ..common.train import train
 
 model.backbone.bottom_up.freeze_at = 2
+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
diff --git a/configs/COCO-Keypoints/keypoint_rcnn_R_50_FPN_1x.py b/configs/COCO-Keypoints/keypoint_rcnn_R_50_FPN_1x.py
@@ -5,3 +5,4 @@
 from ..common.train import train
 
 model.backbone.bottom_up.freeze_at = 2
+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
diff --git a/configs/COCO-PanopticSegmentation/panoptic_fpn_R_50_1x.py b/configs/COCO-PanopticSegmentation/panoptic_fpn_R_50_1x.py
@@ -5,3 +5,4 @@
 from ..common.train import train
 
 model.backbone.bottom_up.freeze_at = 2
+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
diff --git a/configs/common/train.py b/configs/common/train.py
@@ -2,7 +2,7 @@
 # You can use your own instead, together with your own train_net.py
 train = dict(
     output_dir="./output",
-    init_checkpoint="detectron2://ImageNetPretrained/MSRA/R-50.pkl",
+    init_checkpoint="",
     max_iter=90000,
     amp=dict(enabled=False),  # options for Automatic Mixed Precision
     ddp=dict(  # options for DistributedDataParallel
diff --git a/projects/Rethinking-BatchNorm/configs/mask_rcnn_BNhead.py b/projects/Rethinking-BatchNorm/configs/mask_rcnn_BNhead.py
@@ -14,4 +14,5 @@
 optimizer = get_config("common/optim.py").SGD
 train = get_config("common/train.py").train
 
+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
 train.max_iter = 270000  # 3x for batchsize = 16
diff --git a/projects/Rethinking-BatchNorm/configs/retinanet_SyncBNhead.py b/projects/Rethinking-BatchNorm/configs/retinanet_SyncBNhead.py
@@ -10,4 +10,6 @@
 train = get_config("common/train.py").train
 
 optimizer.lr = 0.01
+
+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
 train.max_iter = 270000  # 3x for batchsize = 16

Original file line number	Diff line number	Diff line change
`@@ -5,3 +5,4 @@`
`5`	`5`	`from ..common.models.mask_rcnn_c4 import model`
`6`	`6`
`7`	`7`	`model.backbone.freeze_at = 2`
	`8`	`+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"`
Original file line number	Diff line number	Diff line change
`@@ -5,3 +5,4 @@`
`5`	`5`	`from ..common.train import train`
`6`	`6`
`7`	`7`	`model.backbone.bottom_up.freeze_at = 2`
	`8`	`+train.init_checkpoint = "detectron2://ImageNetPretrained/MSRA/R-50.pkl"`