CCNet-Pure-Pytorch
CCNet-Pure-Pytorch copied to clipboard
Calculating loss
In the ccnet code. At the end of resnet there is out which gives an output of 33X33 sized image. But it is then passed into the self.criterion(outs, labels). But the labels are of original size i.e. 769*769. So, how can it be computed? I am getting the "CUDNN_STATUS_INTERNAL_ERROR" error. I have attached the same in the snapshot below.

@shrutishrestha https://github.com/Serge-weihao/CCNet-Pure-Pytorch/blob/master/loss/criterion.py
@shrutishrestha I remember that the out stride is 8 and the output size is 92?