Talking-Face-Generation-DAVS
Talking-Face-Generation-DAVS copied to clipboard
Unable to save the model using Train.py
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.
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 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.