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

About using the Sentcap dataset and setting parameters for train.py #10

Open
Rouchestand opened this issue Apr 27, 2023 · 5 comments
Open

Comments

@Rouchestand
Copy link

Dear author, I would like to use this code on the Senticap dataset. May I ask for the parameter settings for train.py, such as the number of epochs, the backbone of CLIP, noise_ variance et al

@DavidHuji
Copy link
Owner

DavidHuji commented Apr 27, 2023

Thank you for your interest in our work. Generally, I recommend using the default hyperparameters as a starting point, which are the values defined in the arg-parser. The number of epochs may depend on the size of the dataset, so I suggest training for approximately 20 epochs, but then choose the weights from the epoch where the test error begins to increase.

@Rouchestand
Copy link
Author

Thank you very much for your reply.

@chun178
Copy link

chun178 commented May 5, 2023

Thank you very much for your reply.

Hello, I also need to use the Senticap dataset, but I don't know how to use it. I would like to ask for your advice. Thank you

@Rouchestand
Copy link
Author

You can use the pre trained weights provided by the author, and then use the SentiCap dataset for fine-tuning.

@chun178
Copy link

chun178 commented May 10, 2023

You can use the pre trained weights provided by the author, and then use the SentiCap dataset for fine-tuning.

OK,thank you very much

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

3 participants