如何使用coremltools设置CoreML模型作者？修改后保存不生效

复现步骤 #

temp_model = ct.models.MLModel(current_model_path)

print(f'Original author: {temp_model.author}')
temp_model.author = 'Luka'
print(f'Modified author: {temp_model.author}')
temp_model.save('temp_model.mlpackage')
temp_model = ct.models.MLModel('temp_model.mlpackage')
print(f'After saving: {temp_model.author}')

执行代码后可见，修改后的作者信息在保存并重新加载模型后并未保留。

系统环境 #

• coremltools版本：6.3.0
• 操作系统：MacOS

补充说明 #

我修改的是自己训练并转换为CoreML格式的PyTorch模型。不止作者信息，模型信息面板里的所有元数据都无法通过代码修改。

编辑1：这是Apple已确认的bug，最新beta版本（7.0b1）已修复该问题。

内容的提问来源于stack exchange，提问作者Alcibiades