How can I train the model with custom dataset?

[CreadDiscans]

It looks there is no training code.
Can I get the training code?

[shansongliu]

It looks there is no training code. Can I get the training code?

Have you found the training code?

[chrisdonahue]

Hello! Thanks for the interest in the training code.

I did extensive refactoring and cleaning to polish my ugly research codebase for this publicly-released version. Unfortunately, I never got around to integrating the training code into this refactor, so the scope of this codebase is limited to inference of the pre-trained models.

If you're really eager to dive in and try to get my training code working, it can be found here: https://drive.google.com/file/d/1iI86B5UwoACpdrTET2os4G22DC0hQA24/view?usp=share_link

At this point, it may be easier to start from scratch to retrain the model, using the information in the paper as guidance along with this code I've shared. Unfortunately I won't be able to provide support for this code, but if you can get it working, please consider opening a PR.

[manhdoan291]

@chrisdonahue how to run training and install requirements.txt ? pls, help me!!

[manhdoan291]

@shansongliu @CreadDiscans @vaclavmuller Have you run the training code yet?

[Vosatorp]

Hi! Can you please give a config file for training script?

[119010291]

Hi! Can you please give a config file for training script?

Hi! Have you had the config file?