mirror of https://github.com/JDAI-CV/fast-reid.git
parent
6300bd756e
commit
54f96ba78a
|
@ -60,7 +60,7 @@ class Distiller(Baseline):
|
||||||
for param in model_self.parameters():
|
for param in model_self.parameters():
|
||||||
param.requires_grad_(False)
|
param.requires_grad_(False)
|
||||||
|
|
||||||
if cfg_self.MODEL.WEIGHTS is not '':
|
if cfg_self.MODEL.WEIGHTS != '':
|
||||||
logger.info("Loading self distillation model weights ...")
|
logger.info("Loading self distillation model weights ...")
|
||||||
Checkpointer(model_self).load(cfg_self.MODEL.WEIGHTS)
|
Checkpointer(model_self).load(cfg_self.MODEL.WEIGHTS)
|
||||||
else:
|
else:
|
||||||
|
|
Loading…
Reference in New Issue