MiniCPM-o icon indicating copy to clipboard operation
MiniCPM-o copied to clipboard

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

Open vjaideep08 opened this issue 1 month ago • 0 comments

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

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

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

  • [x] 我已经搜索过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 : https://github.com/modelscope/ms-swift/issues/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

vjaideep08 avatar Jan 21 '25 10:01 vjaideep08