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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training interruption

571129857 opened this issue · comments

problem:
When using 4*1080Ti , training gets stuck, memory is not released, and no error is reported. Forced task termination, retraining is required

environment:
python 3.6
pytorch 1.1
cuda 10.0

code:
branch-1.1.0

I also encountered this problem, and I found it would arise at fixed interval if I continue train.