About training time
Ayanami2019 opened this issue · comments
Hello, thanks for sharing your code. I want to follow up your work but may not have enough equipment. What is the GPU you are using, and how long is the training time on the all data sets?
@Ayanami2019 Thanks for your interests to our work. It works on a GPU with 12GiB memory. It often takes three days to train on the full dataset, but you can also use my pre-trained weights provided in this Github repo to speed up your training.
Thanks for your answer!