PocketNet

PocketNet copied to clipboard

kikirizki

where the pre-trained teacher model?

2

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

xiakj

Why can't my training loss decrease with ArcFace? It drops to about 21 and stops falling.

3

LHH20000923