SE-SSD
SE-SSD copied to clipboard
Change AT_CHECK to TORCH_CHECK
AT_CHECK is deprecated in torch 1.5. Change all the AT_CHECK to TORCH_CHECK.
Thanks for your commit and I appreciate your modification. I'll merge your commit if more people report this problem, considering that many developers haven't updated their torch version to 1.5.
It will be nice if this PR can be merged.