diff --git a/fastreid/modeling/meta_arch/distiller.py b/fastreid/modeling/meta_arch/distiller.py index fc68f3c5..9812143d 100644 --- a/fastreid/modeling/meta_arch/distiller.py +++ b/fastreid/modeling/meta_arch/distiller.py @@ -60,7 +60,7 @@ def __init__(self, cfg): for param in model_self.parameters(): param.requires_grad_(False) - if cfg_self.MODEL.WEIGHTS is not '': + if cfg_self.MODEL.WEIGHTS != '': logger.info("Loading self distillation model weights ...") Checkpointer(model_self).load(cfg_self.MODEL.WEIGHTS) else: