SongTang-x/SwinLSTM

How long does a complete training take?

Opened this issue · 1 comments

zpeng2022 commented

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 commented

  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.