CenterNet on pytorch>=1.7 can't works

I need to use CenterNet on pytorch>=1.7 but it can't works, it only works on pytorch<=1.4, but I truly need it, what can I do?