before merge

configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py

@@ -66,15 +66,16 @@ optimizer = dict(_delete_=True, type='AdamW', lr=0.0001, betas=(0.9, 0.999), wei
                                                  'relative_position_bias_table': dict(decay_mult=0.),
                                                  'norm': dict(decay_mult=0.)}))
 lr_config = dict(step=[8, 11])
-runner = dict(type='EpochBasedRunnerAmp', max_epochs=12)
+#runner = dict(type='EpochBasedRunnerAmp', max_epochs=12)
+runner = dict(type='EpochBasedRunner', max_epochs=12)
 
 # do not use mmdet version fp16
-fp16 = None
-optimizer_config = dict(
-    type="DistOptimizerHook",
-    update_interval=1,
-    grad_clip=None,
-    coalesce=True,
-    bucket_size_mb=-1,
-    use_fp16=True,
-)
+# fp16 = None
+# optimizer_config = dict(
+#     type="DistOptimizerHook",
+#     update_interval=1,
+#     grad_clip=None,
+#     coalesce=True,
+#     bucket_size_mb=-1,
+#     use_fp16=True,
+# )

tools/gen_checkpoint.py

@@ -23,8 +23,8 @@ def get_sd(filename, return_sd=True):
 
 def merge(target, backbone, head):
     tsd = target['state_dict']
-    bsd = target['state_dict']
-    hsd = target['state_dict']
+    bsd = backbone['state_dict']
+    hsd = head['state_dict']
     for key in tsd.keys():
         if 'backbone' in key:
             assert key in bsd
@@ -41,11 +41,11 @@ def main():
     backbone = get_sd(args.backbone, return_sd=False)
     head = get_sd(args.head, return_sd=False)
 
-    target = backbone.copy()
-    #target = head.copy()
+    #target = backbone.copy()
+    target = head.copy()
 
     target = merge(target, backbone, head)
-    os.makedirs(os.path.basename(args.out), exist_ok=True)
+    #os.makedirs(os.path.basename(args.out), exist_ok=True)
     torch.save(target, args.out)
     print("saved checkpoint in {}".format(args.out))