Thank you for code sharing! I guess you could change the line 8 in train.py to 'from encodegan import EncodeGAN', then this project will be perfekt.