HRNet / HRNet-Semantic-Segmentation

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919
Other
3.13k stars 686 forks source link

About the training error #239

Open JunkunPeng17 opened 3 years ago

JunkunPeng17 commented 3 years ago

Dear author, when I try to train the model in RTX3090, an issue is that it could just be worked when I use 2 GPUs. If I use 4 GPUs, The server will automatically restart after training a few iterations. I wonder if you know some suggestions.