CCNet-Pure-Pytorch icon indicating copy to clipboard operation
CCNet-Pure-Pytorch copied to clipboard

Calculating loss

Open shrutishrestha opened this issue 4 years ago • 2 comments

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.

ccnet

shrutishrestha avatar Feb 07 '21 11:02 shrutishrestha

@shrutishrestha https://github.com/Serge-weihao/CCNet-Pure-Pytorch/blob/master/loss/criterion.py

Serge-weihao avatar Feb 07 '21 11:02 Serge-weihao

@shrutishrestha I remember that the out stride is 8 and the output size is 92?

Serge-weihao avatar Feb 07 '21 11:02 Serge-weihao