search

Yunfan-Li / Twin-Contrastive-Learning

Code for the paper "Twin Contrastive Learning for Online Clustering" (IJCV 2022)
56 stars 11 forks source link

About training time and dataloader of ImageNet-10 and ImageNet-dogs #14

Closed kasteric closed 1 year ago

kasteric commented 1 year ago

Hi, thanks for your awesome work! I am wondering how much time do u take to train ImageNet-10 for 1000 epochs. When I was training ImageNet-10 using ImageFolder, the training speed, especially the I/O speed of dataloader was very slow, do you have the same issues？

Yunfan-Li commented 1 year ago

Hi, it takes about 14 hours to train ImageNet-10 on a 3090 GPU. You could try different num_worker value of dataloader.

kasteric commented 1 year ago

Thanks for your kind response!