Unable to save the model using Train.py
AtaUllahB opened this issue · comments
AtaUllahB. commented
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.
Mombin commented
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