Hi Everyone,
I am pretty new to pytorch and I am trying to implement semantic segmentation using
torchvision.models.segmentation. fcn_resnet50 as mentioned on https://pytorch.org/docs/stable/torchvision/models.html#semantic-segmentation 34
While running my code, I receive the following error:
‘collections.OrderedDict’ object has no attribute ‘log_softmax’
This correspond to the line in the code —> 10     loss = criterion(outputs, labels)
I defined my criterion loss function as criterion = nn.CrossEntropyLoss()
Any help is appreciated!!!
Thanks
Troy