search

sithu31296 / semantic-segmentation

SOTA Semantic Segmentation Models in PyTorch
MIT License
852 stars 152 forks source link

pytorch version >= 1.9 ? #40

Open hirokic5 opened 2 years ago

hirokic5 commented 2 years ago

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 .

scl666 commented 2 years ago

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