'FaceswapGANModel' object has no attribute 'netDA_train'

[yangdazhuo816]

AttributeError Traceback (most recent call last)
in
112 data_A = train_batchA.get_next_batch()
113 data_B = train_batchB.get_next_batch()
--> 114 errDA, errDB = model.train_one_batch_D(data_A=data_A, data_B=data_B)
115 errDA_sum +=errDA[0]
116 errDB_sum +=errDB[0]

F:\faceswap-GAN-master\networks\faceswap_gan_model.py in train_one_batch_D(self, data_A, data_B)
326 else:
327 raise ValueError("Something's wrong with the input data generator.")
--> 328 errDA = self.netDA_train([warped_A, target_A])
329 errDB = self.netDB_train([warped_B, target_B])
330 return errDA, errDB

AttributeError: 'FaceswapGANModel' object has no attribute 'netDA_train'

How can I solve this problem

[alvinahmadov]

Solution: Add self.netDA_train = None, self.netDB_train = None, etc in __init__ in faceswap_gan_model.py