mlvlab/SpeaQ

Training initial weight for SpeaQ

fujiawei0724 opened this issue · 1 comments

Thanks for the great work and code base. I am curious about how you train the initial weight (detr weight to initialize speaq model). When I want to train it by myself, it reports this bug:
Screenshot 2024-08-04 at 21 44 47
It seems that the training process is unstable, I want to know how to train the model and get this initial weight for detr. Thanks for your help, looking forward to your reply.

Hi, thanks for your interest in our work.

As mentioned in the README, we utilized the pre-trained DETR weights provided in this repository rather than training the model ourselves.
Therefore, for detailed information about the training process, it would be best to consult the original authors.
Additionally, you may find relevant information in this issue thread of the original repository.