PyTorch-Encoding 训练问题

训练问题

Open lkhazyf opened this issue 4 years ago • 4 comments

我在训练是实话每次都会卡在运行的过程中，train.py里面outputs = self.model(image)这一步走不通。此外，我设置的pretrained=false，请问为什么每次都会下载pretrained模型呢？请问这两个问题该怎么解决啊，谢谢！

Nov 27 '20 01:11 lkhazyf

我的训练集是citys

Nov 27 '20 01:11 lkhazyf

这个代码单个 gpu 可能会卡住，你用了几个gpu？

Dec 08 '20 06:12 zhanghang1989

我用了两块3090也遇到了同样的问题，起初跑通过一次，然后就一直卡在最开始

Dec 08 '21 06:12 nonword1

用的 train_dist.py 这个 script 吗？这个速度也会快很多 https://github.com/zhanghang1989/PyTorch-Encoding/blob/master/experiments/segmentation/train_dist.py

另外，这个 package 已经很老了，我自己已经几年没用了，建议使用 mmsegmentation

Dec 09 '21 00:12 zhanghang1989