search

NVlabs / GroupViT

Official PyTorch implementation of GroupViT: Semantic Segmentation Emerges from Text Supervision, CVPR 2022.
https://arxiv.org/abs/2202.11094
Other
725 stars 52 forks source link

single GPU #38

Open scm-later opened 2 years ago

scm-later commented 2 years ago

Hello, can the model be trained on a single GPU

xvjiarui commented 2 years ago

Hi @scm-later

The model trained on single GPU may not have good accuracy since the contrastive learning needs large batch size.