fix typo

MCG-NJU · Mar 30, 2023 · 5532478 · 5532478
1 parent ea4a311
commit 5532478
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/lib/models/mixformer_convmae/mixformer.py b/lib/models/mixformer_convmae/mixformer.py
@@ -408,7 +408,7 @@ def get_mixformer_convmae(config, train):
 
     if config.MODEL.BACKBONE.PRETRAINED and train:
         ckpt_path = config.MODEL.BACKBONE.PRETRAINED_PATH
-        ckpt = torch.load(ckpt_path, map_location='cpu') #['model']
+        ckpt = torch.load(ckpt_path, map_location='cpu')['model']
         new_dict = {}
         for k, v in ckpt.items():
             if 'pos_embed' not in k and 'mask_token' not in k:

diff --git a/lib/models/mixformer_convmae/mixformer_online.py b/lib/models/mixformer_convmae/mixformer_online.py
@@ -409,7 +409,7 @@ def get_mixformer_convmae(config, train):
 
     if config.MODEL.BACKBONE.PRETRAINED and train:
         ckpt_path = config.MODEL.BACKBONE.PRETRAINED_PATH
-        ckpt = torch.load(ckpt_path, map_location='cpu') #['model']
+        ckpt = torch.load(ckpt_path, map_location='cpu')['model']
         new_dict = {}
         for k, v in ckpt.items():
             if 'pos_embed' not in k and 'mask_token' not in k:

diff --git a/lib/test/tracker/mixformer_convmae.py b/lib/test/tracker/mixformer_convmae.py
@@ -12,7 +12,7 @@
 class MixFormer(BaseTracker):
     def __init__(self, params, dataset_name):
         super(MixFormer, self).__init__(params)
-        network = build_mixformer_convmae(params.cfg)
+        network = build_mixformer_convmae(params.cfg, train=False)
         network.load_state_dict(torch.load(self.params.checkpoint, map_location='cpu')['net'], strict=True)
         self.cfg = params.cfg
         self.network = network.cuda()

diff --git a/lib/test/tracker/mixformer_cvt.py b/lib/test/tracker/mixformer_cvt.py
@@ -12,7 +12,7 @@
 class MixFormer(BaseTracker):
     def __init__(self, params, dataset_name):
         super(MixFormer, self).__init__(params)
-        network = build_mixformer_cvt(params.cfg)
+        network = build_mixformer_cvt(params.cfg, train=False)
         network.load_state_dict(torch.load(self.params.checkpoint, map_location='cpu')['net'], strict=True)
         self.cfg = params.cfg
         self.network = network.cuda()

diff --git a/lib/test/tracker/mixformer_vit.py b/lib/test/tracker/mixformer_vit.py
@@ -12,7 +12,7 @@
 class MixFormer(BaseTracker):
     def __init__(self, params, dataset_name):
         super(MixFormer, self).__init__(params)
-        network = build_mixformer_vit(params.cfg)
+        network = build_mixformer_vit(params.cfg, train=False)
         network.load_state_dict(torch.load(self.params.checkpoint, map_location='cpu')['net'], strict=True)
         self.cfg = params.cfg
         self.network = network.cuda()