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

Code for Talking Face Generation by Adversarially Disentangled Audio-Visual Representation (AAAI 2019)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Unable to save the model using Train.py

AtaUllahB opened this issue · 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

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.