I found this guy: DiscoGAN/image_translation.py at master · ptrblck/DiscoGAN · GitHub 7
Instead could you write this as a class Train(model) where in the constructor self.args = args just gets the arguments from argparse and you can then write a method that uses a similar training loop to the one you have? model can be any of the objects in model.py.
I am trying to do this, and keep getting an attributerror after calling an instance of my model using super().__init__(args) within the constructor of my Train object…