search

QData / C-Tran

General Multi-label Image Classification with Transformers
MIT License
238 stars 40 forks source link

Optimal no. of epochs to train a model and availability of a trained model #17

Open ans92 opened 1 year ago

ans92 commented 1 year ago

Hi, Thank you for the code and the great work. I have two questions regarding your work. First how many number of epochs your model needs to train successfully? I am running your model on my dataset however it is not learning till 10 epochs therefore I want to know how many epochs your model needs. Secondly do you also have trained models as transformers need a lot of resources for training. Therefore if you can provide trained models that would help me a lot. Thank you for your time.

Q418 commented 1 year ago

Hello, I am trying to run this model on my own dataset, but the results after 100 epochs of training are similar to the beginning, it seems that it have not learned anything, so I would like to ask you if this is the case?

ans92 commented 1 year ago

I tested the code on a small dataset and appears that it works. But the accuracy is low. You need to check your data to ensure you are not making any mistake.

jacklanchantin commented 1 year ago

Hi, Thank you for the code and the great work. I have two questions regarding your work. First how many number of epochs your model needs to train successfully? I am running your model on my dataset however it is not learning till 10 epochs therefore I want to know how many epochs your model needs. Secondly do you also have trained models as transformers need a lot of resources for training. Therefore if you can provide trained models that would help me a lot. Thank you for your time.

How large is your dataset?

Sorry, I don't have access to the trained models anymore. Do you have access to a GPU?