Skip to content

【开源实习】GIT模型应用开发 #2027

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

wujiu0411
Copy link

第一个自定义数据集上微调GIT模型进行图像描述的笔记本中,我将在一个小型图像描述数据集上微调

这是在Mindspore的训练过程图,50个epoch后loss稳定在0.06左右
ms训练图
这是在pytorch的训练过程图,50个epoch后loss稳定在0.09左右
torch训练图
下面是mindspore的推理图,可以看到推理结果与预期一致
ms推理结果图
下面是pytorch的推理图,可以看到准确度相较于mindspore差一点
torch推理结果图

第二个使用 GIT 进行图像/视频描述生成和图像/视频问答的笔记本中,演示了如何使用MindSpore的 GIT 模型来对图像或视频进行描述生成,以及在图像或视频上进行问答。

选择其中一个推理模型进行比较:
mindspore:
ms推理2
pytorch:
orch推理2
结果一致

@wujiu0411
Copy link
Author

任务

任务编号:#IAADHU
任务链接【开源实习】GIT模型应用开发
实现内容:实现了自定义数据集上微调GIT模型进行图像描述以及使用 GIT 进行图像/视频描述生成和图像/视频问答。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant