Imported from https://github.com/artix41/ALFI-pytorch