Closed zhenyuan1234 closed 1 year ago
According to the configuration in the paper, it takes 4 days to train on 2 3090s, but I have a faster configuration for experiments, and it takes about 1 day. If necessary, you can contact my email address zdw1999@zju.edu.cn to apply for a complete training code.
Hello Can I get the faster configuration for experiments? Now I am working on the condition of gpu=1, batch size=4, May I ask how many days it would take to retrain your model? Also can I get the FPS code too? Looking for ur reply! My email: finn951017@outlook.com
@finnn00 To enable PyDiff to train on a single GPU, the PyDiff/options/train_v2.yaml file can be consulted.
How long did it take to train the model?