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

[541607120115]

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'