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

The result of training is very poor. #11

Open
hahahaa1 opened this issue May 6, 2024 · 2 comments
Open

The result of training is very poor. #11

hahahaa1 opened this issue May 6, 2024 · 2 comments

Comments

@hahahaa1
Copy link

hahahaa1 commented May 6, 2024

According to the requirements in the paper, under the conditions of batch_size = 8 and epochs = 150, the result trained in the train.py file is very poor. There is no default training weight file in the train.py file. I don't know if this is the reason why my training result is very unsatisfactory. Could the author please answer me?
微信图片_20240506135833
微信图片_20240506135905

@mkang315
Copy link
Owner

mkang315 commented May 6, 2024

Precisely, the training result is meager. I cannot judge what's wrong with your settings only by the screenshot.

@hahahaa1
Copy link
Author

hahahaa1 commented May 6, 2024

I am reproducing on the Windows system. In the train.py file, I have made the following settings: batch_size = 8 and epochs = 150, but the result is very unsatisfactory. I would like to ask if you used pre-trained weights when conducting the training. My training result is really a headache. Thank you for your answer.
Uploading 微信图片_20240506192005.png…

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

2 participants