roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

stucked when training is done #198

Open greathope opened 5 years ago

greathope commented 5 years ago

pytorch0.4.1 cuda9+cudnn7

After saving last step checkpoint, detectron just got stucked. I manually killed it, the error infos shows it may be the deadlock in dataloader, can anyone help me?