Talking-Face-Generation-DAVS icon indicating copy to clipboard operation
Talking-Face-Generation-DAVS copied to clipboard

Unable to save the model using Train.py

Open AtaUllahB opened this issue 5 years ago • 2 comments

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

AtaUllahB avatar Oct 19 '19 08:10 AtaUllahB

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

Mombin avatar Nov 11 '19 08:11 Mombin

@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.

Hangz-nju-cuhk avatar Nov 18 '19 11:11 Hangz-nju-cuhk