NbNet-Pytorch src:https://github.com/csgcmai/NBNet Rewrite from tensorflow to pytorch