wildBigPanda

@XuanYang-cn Is the design unreasonable? To drop partition, you have to call the release interface again separately, and the related interface for releasing partitions cannot be found in [collection.py]. How...

style_encoder = StyleEncoder(config) model_CKPT = torch.load(config.style_encoder_ckpt, map_location="cpu") model_ckpt = {} for key, value in model_CKPT['model'].items(): new_key = key[7:] model_ckpt[new_key] = value style_encoder.load_state_dict(model_ckpt, strict=False) 这段代码，key=‘emotion_clf.classifier.bias’ 和key = ‘emotion_clf.classifier.weight’ ，只有7个维度的情感类型