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

Paper reproduction #7

Open
xuliuwei opened this issue Nov 16, 2023 · 7 comments
Open

Paper reproduction #7

xuliuwei opened this issue Nov 16, 2023 · 7 comments

Comments

@xuliuwei
Copy link

I recently tried to reproduce the results of your paper, but I was unable to achieve the results of the paper. I trained with two 3090 students and batchsize of 12, and the learning rate is the same as that of the paper. May I ask whether the learning rate is the reason for the poor result? Does the learning rate have a big impact on your result? Can you provide your pre-training model?

@lhaippp
Copy link

lhaippp commented Nov 21, 2023

Did you leverage the python environments in this repo?

pip install -r requirements.txt

@xuliuwei
Copy link
Author

I'll try to change the environment. Another problem is that I found that the training parameter Settings in the paper are different from the parameter Settings in the code, which one should I use?
image
image

@lhaippp
Copy link

lhaippp commented Nov 22, 2023

I recommend you to use the experiment setting in this repo

@xuliuwei
Copy link
Author

Did you run it with four graphics cards?

@lhaippp
Copy link

lhaippp commented Nov 22, 2023

I think the number of GPU cards is not important. As long as you can run the code with the setting

@xuliuwei
Copy link
Author

How long does it take you to train an epoch? I used two 3090s and it took 3.5 hours to train an epoch.

@six-finger
Copy link

@xuliuwei May I chat with you about this issue? Thank you~ Wechat: margin333

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