DeTr 保存和导出自定义训练模型

Question

我在训练模型后下载的文件：

1. 配置.json
2. 模型.安全张量
3. 日志 > hparams/yaml（空文件）
4. 日志 > events.out.tfevents.1707970167.d4c7dd713d7a.235.0
5. 日志 > 检查点 > 输出.ckpt

---

以下代码用于保存：

MODEL_PATH = 'custom-model'
model.model.save_pretrained(MODEL_PATH)

---

自定义模型文件夹包含文件号1和2（config.json和model.safetensors）

当我尝试加载我在 google colab 的同一会话中自定义训练的模型时，它工作正常，但是当将同一文件夹复制到另一个 colab 笔记本并尝试执行时，它会抛出此错误：

/data does not appear to have a file named preprocessor_config.json

---

此代码用于加载自定义模型

CHECKPOINT = 'models/data'

image_processor = DetrImageProcessor.from_pretrained(CHECKPOINT)
model = DetrForObjectDetection.from_pretrained(CHECKPOINT)

Answer 1

好吧，问题出在这一行

image_processor = DetrImageProcessor.from_pretrained(CHECKPOINT)

将检查点替换为

'facebook/detr-resnet-50'

加载模型的完整代码

CHECKPOINT = 'models/data'

image_processor = DetrImageProcessor.from_pretrained('facebook/detr-resnet-50') 
model = DetrForObjectDetection.from_pretrained(CHECKPOINT)

模型/数据包含上述问题中提到的所有文件（1-5）