search

nasir6 / zero_shot_detection

61 stars 18 forks source link

Training Time #7

Closed AmingWu closed 3 years ago

AmingWu commented 3 years ago

Nice work.

How long does your model need to train?

nasir6 commented 3 years ago

It roughly takes about 8~9 hours on a machine with 8 GPUs to train the detector on seen classes. and then about one hour to extract and save the features, then about 1-2 hours to train the generator. So in total, it takes about 12 hours for the MS-COCO dataset.