search

YanjingLi0202 / Q-ViT

The official implementation of the NeurIPS 2022 paper Q-ViT.
83 stars 7 forks source link

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

Open charliezjw opened 2 years ago

charliezjw commented 2 years ago

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!

YanjingLi0202 commented 2 years ago

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

charliezjw commented 2 years ago

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