No module named 'models'

[dvd42]

Hi, thanks for providing the code for your paper.

Trying to get started with the project I ran into an issue when running the run_gen.sh script inside the IntervenedGAN folder I get:

File "gen.py", line 13, in <module>
    from models import get_instrumented_model
ModuleNotFoundError: No module named 'models'

I looked around the project and I couldn't find any module named 'models' that contained the function get_instrumented_model. Is this a bug or did I miss something?

Note that the BigGan/single_gen.py script works just fine.

Thanks in advance for any help!!