search

Hangz-nju-cuhk / Talking-Face-Generation-DAVS

Code for Talking Face Generation by Adversarially Disentangled Audio-Visual Representation (AAAI 2019)
MIT License
817 stars 173 forks source link

Unable to save the model using Train.py #35

Open AtaUllahB opened 4 years ago

AtaUllahB commented 4 years ago

To save the network we are using a new python file net.py which has the code of Train.py up till the code: model = Gen_Model.GenModel(opt) after which we are trying to save the model using torch.save() but the class is importing Gen_final_v1 which imports embedding_utils which causes the following error. Kindly help. name

Mombin commented 4 years ago

modify your code in embedding_utils.py origin: import Options config = Options.Config() revise: from Options import Config config = Config().parse()

and if you find same error in another code, you can fix same way

Hangz-nju-cuhk commented 4 years ago

@Mombin Thank you very much! As you can see that I actually wrote the config file into a class, and changed it to argparse when releasing the code.