未验证 提交 c61c68d4 编写于 作者: L LokeZhou 提交者: GitHub

fix trainer.py load_weights_sde when reid mode is None (#7604)

* fix trainer.py load_weights_sde when reid mode is None

* fix trainer.py load_weights_sde
上级 de39e703
...@@ -399,7 +399,8 @@ class Trainer(object): ...@@ -399,7 +399,8 @@ class Trainer(object):
def load_weights_sde(self, det_weights, reid_weights): def load_weights_sde(self, det_weights, reid_weights):
if self.model.detector: if self.model.detector:
load_weight(self.model.detector, det_weights) load_weight(self.model.detector, det_weights)
load_weight(self.model.reid, reid_weights) if self.model.reid:
load_weight(self.model.reid, reid_weights)
else: else:
load_weight(self.model.reid, reid_weights) load_weight(self.model.reid, reid_weights)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册