Pytorch implementation of face attention network