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.

Sign up for GitHub

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

Jump to bottom

train_one_epoch() is pass undefined parameter in main.py #3

Open

charliezjw opened this issue Nov 8, 2022 · 2 comments

charliezjw commented Nov 8, 2022

Hi, it's a very nice paper with excellent results. But I am trying to train the model by modifying your code, I notice https://github.com/charliezjw/Q-ViT/blob/main/main.py#L383 is passing a parameter that is not defined in https://github.com/charliezjw/Q-ViT/blob/cc10d670dad579d937c6821829fc1acc201033b4/engine.py#L19 .

I am wondering whether there is a different main.py file for training.

Thank you very much!

Owner

YanjingLi0202 commented Nov 8, 2022

Sorry for the bug, we have fixed it in the new commit. The full training code will be released soon.

Author

charliezjw commented Nov 8, 2022

Great, looking forward to it. Is there an estimated time for that?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment