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

coco-hand and coco basic training command difference #1

Closed
YoungjaeDev opened this issue May 29, 2022 · 3 comments
Closed

coco-hand and coco basic training command difference #1

YoungjaeDev opened this issue May 29, 2022 · 3 comments

Comments

@YoungjaeDev
Copy link

https://github.com/uyzhang/yolov5_prune#steps

In basic training number 2, coco-hand is based on pre-trained weight, but it is not coco, which one is correct?
As far as I know, it is related to the botleneck ignore problem for sparsity training, should we proceed with the existing pretrained weight before sparsity training?

@uyzhang
Copy link
Owner

uyzhang commented May 30, 2022

It's been modified. The basic training of data sets does not need to load pre-training weights. thx!

@YoungjaeDev
Copy link
Author

@uyzhang is there any reason why you did it with adamw? There is not much difference from sgd to calc sparsity, but if there is another reason, can I know?

@uyzhang
Copy link
Owner

uyzhang commented May 30, 2022

I tried sparse training with SGD, but the effect was not good. AdamW works well, so I end up using AdamW.

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