sithu31296 / semantic-segmentation

SOTA Semantic Segmentation Models in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch version >= 1.9 ?

hirokic5 opened this issue · comments

Hi, thank you for great repository !!!

When I tested pretrained model, there was error in this line.

@torch.inference_mode()

@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 .

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