SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023
https://praeclarumjj3.github.io/oneformer
MIT License
1.41k stars 128 forks source link

gpu cost duing training #30

Closed JunMa11 closed 1 year ago

JunMa11 commented 1 year ago

Thanks for sharing the awesome work!

I have a minor question.

how many GPU hours does the model training process need on the COCO and ADE20K datasets?

praeclarumjj3 commented 1 year ago

Hi @JunMa11, thanks for your interest in our work.

For Swin-L OneFormer, the training times are roughly as follows:

JunMa11 commented 1 year ago

Thanks for your reply very much.