报错:RuntimeError: Function CdistBackward returned an invalid gradient at index 1 – got [2,7,7] but expected shape compatible with [2, 2048, 7, 7]
原因:调用Pytorch.cdist().sum() 计算Loss是有问题的,
正确的Lorm范数计算为torch.dist或者torch.norm(x-y);
报错:RuntimeError: Function CdistBackward returned an invalid gradient at index 1 – got [2,7,7] but expected shape compatible with [2, 2048, 7, 7]
原因:调用Pytorch.cdist().sum() 计算Loss是有问题的,
正确的Lorm范数计算为torch.dist或者torch.norm(x-y);