Ikaros521/moe-tts

1import torch2 3if __name__ == '__main__':4    model_path = "saved_model/18/model.pth"5    output_path = "saved_model/18/model1.pth"6    checkpoint_dict = torch.load(model_path, map_location='cpu')7    checkpoint_dict_new = {}8    for k, v in checkpoint_dict.items():9        if k == "optimizer":10            print("remove optimizer")11            continue12        checkpoint_dict_new[k] = v13    torch.save(checkpoint_dict_new, output_path)14