Skip to content
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

[BUG] <title> Creation of Product Categorization model by fine tuning MiniCPM-V-2_6 model on e-commerce product data #773

Open
2 tasks done
vjaideep08 opened this issue Jan 21, 2025 · 0 comments

Comments

@vjaideep08
Copy link

是否已有关于该错误的issue或讨论? | Is there an existing issue / discussion for this?

  • 我已经搜索过已有的issues和讨论 | I have searched the existing issues / discussions

该问题是否在FAQ中有解答? | Is there an existing answer for this in FAQ?

  • 我已经搜索过FAQ | I have searched FAQ

当前行为 | Current Behavior

Hi,
I am trying to create a finetuned model for Product<>Category Prediction from MiniCPM-V-2_6 model.
Objective/ Problem statement : Prediction of category from the set of 1000+ categories when given an input product image. We have a set of 1000+ categories, where we need to predict the best category for a given input product image where the product in the image can be mapped with.

What has been done :
I used Swift Framework for finetuning the model on 4 lakh+ dataset. Link : modelscope/ms-swift#1613

期望行为 | Expected Behavior

No response

复现方法 | Steps To Reproduce

No response

运行环境 | Environment

- OS:
- Python:
- Transformers:
- PyTorch:
- CUDA (`python -c 'import torch; print(torch.version.cuda)'`):

备注 | Anything else?

No response

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

No branches or pull requests

1 participant