semantic-segmentation
semantic-segmentation copied to clipboard
pytorch version >= 1.9 ?
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 .
你好,请问有使用camvid数据集吗,我使用DDRNet训练过程中损失出现nan,cityscapes数据集没有这个错误