Chinese-CLIP 关于自己训练数据集模型推理的问题

ff29svar  于 2个月前  发布在  其他
关注(0)|答案(1)|浏览(24)

我基于clip_cn_vit-h-14.pt模型训练了一个自己的模型,并使用下面的代码对模型进行了保存

for epoch in range(num_epochs):
    total_loss = 0
    total_loss2 = 0

但是我不知道该如何推理他,谁能告诉我,谢谢。

tzcvj98z

tzcvj98z1#

请参考预测及评估部分的流程,将 resume 替换为保存的模型路径。

相关问题