PocketNet icon indicating copy to clipboard operation
PocketNet copied to clipboard

where the pre-trained teacher model?

Open xiakj opened this issue 3 years ago • 2 comments

`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.

xiakj avatar May 17 '22 06:05 xiakj

You can use the pretrained model from ElasticFace

fdbtrs avatar May 24 '22 14:05 fdbtrs

You can use the pretrained model from ElasticFace @fdbtrs I have download the pretrained model from ElasticFace, but it seems like the embedding_size of these model are 512. Could you please provide the 128d version? or where could i find these pretrained model

soulyiran avatar May 30 '22 10:05 soulyiran