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

request for yolof-dc5 model #28

Open
emjay73 opened this issue May 13, 2021 · 4 comments
Open

request for yolof-dc5 model #28

emjay73 opened this issue May 13, 2021 · 4 comments

Comments

@emjay73
Copy link

emjay73 commented May 13, 2021

Can I get a model config and pre-trained weights of the model denoted in this table?
image

@chensnathan
Copy link
Collaborator

Hi, we do not have a plan to provide the 3-rd stage model now. Maybe next month. I'll keep this issue open.

You can train the third stage on your own with the provided 2-nd stage pre-trained model. Please follow the steps in this repo.

@emjay73
Copy link
Author

emjay73 commented May 13, 2021

You mean the only difference between the YOLOF_CSP_D_53_DC5_9x_stage2_3x and YOLOF-DC5 in the above table is whether the SWA phase is applied or not? Then where does the fps difference come from?

@chensnathan
Copy link
Collaborator

You can check the caption of the Table. When comparing with YOLOv4, we first fuse conv and bn, then keep cudnn_benchmark as True, and use half-precision for inference.

@emjay73
Copy link
Author

emjay73 commented May 31, 2021

Which means,

SOLVER.AMP.ENABLED : true 
CUDNN_BENCHMARK : true

in the config file?

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