train_one_epoch() is pass undefined parameter in main.py

Question

charliezjw opened this issue 2 years ago · 2 comments

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!

Answer 1 · 2022-11-08T03:54:49.000Z

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

Answer 2 · 2022-11-08T04:00:06.000Z

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