gradcheck for Sqrtm in MPNCOV.py

jiangtaoxie / fast-MPN-COV

@CVPR2018: Efficient unrolling iterative matrix square-root normalized ConvNets, implemented by PyTorch (and code of B-CNN,Compact bilinear pooling etc.) for training from scratch & finetuning.

http://peihuali.org/iSQRT-COV/index.html

MIT License

270 stars 56 forks source link

gradcheck for Sqrtm in MPNCOV.py #28

Open soarodo opened 3 years ago

soarodo commented 3 years ago

Hello, I use autograd.gradcheck for Sqrtm in MPNCOV.py and the function returns false, but if I delete the ligne 'der_NSiter = der_NSiter.transpose(1, 2)' , it turns out to be true.

Is there anybody who could explain it?

Thanks so much