-
Notifications
You must be signed in to change notification settings - Fork 10
Open
Description
`backbone_teacher = iresnet100(num_features=cfg.embedding_size).to(local_rank)
try:
backbone_teacher_pth = os.path.join(cfg.teacher_pth, str((epoch + 1)*11372) + "backbone.pth")
backbone_teacher.load_state_dict(torch.load(backbone_teacher_pth, map_location=torch.device(local_rank)))
if rank == 0:
logging.info("backbone teacher loaded for epoch {} successfully!".format(epoch))
except (FileNotFoundError, KeyError, IndexError, RuntimeError):
logging.info("load teacher backbone for epoch {} init, failed!".format(epoch))
break`
I can not find the teacher model.
Metadata
Metadata
Assignees
Labels
No labels