theEricMa / OTAvatar

This is the official repository for OTAvatar: One-shot Talking Face Avatar with Controllable Tri-plane Rendering [CVPR2023].
306 stars 37 forks source link

Question about 'FaceTrainer' object has no attribute 'net_G_module' #19

Open 541607120115 opened 11 months ago

541607120115 commented 11 months ago

I'm getting an error after loading the model during testing load [net_Warp] and [net_Warp_ema] from result\otavatar\epoch_00005_iteration_000002000_checkpoint.pt Done with loading the checkpoint. 0%| | 0/3537 [00:00<?, ?it/s] 0%| | 0/19 [00:42<?, ?it/s] Traceback (most recent call last): File "inference_refine_1D_cam.py", line 166, in opt_Ws, w_opt, w_std = trainer.inverse_setup(1,) File "F:\00Liss\01mycode\09Voice_driven_face_generation\03code\39-OTAvatar-main\trainers\decouple_by_invert.py", line 600, in inverse_setup w_avg, w_std = self.sample_zs() File "F:\00Liss\01mycode\09Voice_driven_face_generation\03code\39-OTAvatar-main\trainers\decouple_by_invert.py", line 766, in sample_zs self.net_G_module.z_dim AttributeError: 'FaceTrainer' object has no attribute 'net_G_module'