semantic-segmentation icon indicating copy to clipboard operation
semantic-segmentation copied to clipboard

pytorch version >= 1.9 ?

Open hirokic5 opened this issue 2 years ago • 1 comments

Hi, thank you for great repository !!!

When I tested pretrained model, there was error in this line. https://github.com/sithu31296/semantic-segmentation/blob/492231c0d77088267ae6ce6b21a4107ccf8e92f8/tools/infer.py#L71

@torch.inference_mode() method is implemented in torch version 1.9. later.

So, I would be appreciate if you could change torch version in README from >= 1.8,1 to >= 1.9 .

hirokic5 avatar Jul 05 '22 04:07 hirokic5

你好,请问有使用camvid数据集吗,我使用DDRNet训练过程中损失出现nan,cityscapes数据集没有这个错误

scl666 avatar Oct 09 '22 09:10 scl666