Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

How long does a complete training take? #4

Open
zpeng2022 opened this issue Nov 5, 2023 · 1 comment
Open

How long does a complete training take? #4

zpeng2022 opened this issue Nov 5, 2023 · 1 comment

Comments

@zpeng2022
Copy link

zpeng2022 commented Nov 5, 2023 •
edited
Loading

Thank you for your work.

How long does a complete training on MNIST dataset take?

  • I'm training the model using an RTX 3090 GPU with the original config file.

Can you provide a dataloader for the taxiBJ dataset?

  • thank you in advance.
@SongTang-x
Copy link
Owner

  1. The specific training duration depends on your hardware and the number of epochs preset. Training for 2000 epochs on an RTX A5000 takes approximately one week. I hope this serves as a reference for you.

  2. For the dataloader of the taxiBJ dataset, please refer to https://github.com/Yunbo426/MIM/blob/master/src/data_provider/taxibj.py.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@SongTang-x @zpeng2022