Imported from https://github.com/Guillem96/zsl-pytorch