LittleDijkstraZ

你好，我想请问一下repo底下的权重（比如 CULane 的几个backbones）是否包含 CLRnet 的 head 的权重呢？ 出于特殊需求，我通过registry里面的 build_net 来调用 CLRNet 模型，然后想通过正常的 load_state_dict 的方式 加载 你们提供的pretrained weights。不过当我用torch.load 加载这些backbone之后，与CLRnet 模型对比，发现有4个对不上，它们分别是： module.heads.sample_x_indexs module.heads.prior_feat_ys module.heads.prior_ys module.heads.criterion.weight 而其他的weights也能成功地加载进模型，因此我的疑惑是，CLRnet 的这4个weights是不是不在这些 backbone里面？除了重新训练，是否有别的办法获得这些权重呢？谢谢！