Imported from https://github.com/ceyzaguirre4/mac-network-pytorch