add

app.py

@@ -118,24 +118,12 @@ class BaseTrainer(object):
         logger.info(f"Init test dataloader success")
 
         from models.denoiser import MDM
-        self.model = MDM(args).cuda()
+        self.model = MDM(args)
         
         if self.rank == 0:
             logger.info(self.model)
             logger.info(f"init {args.g_name} success")
 
-        self.smplx = smplx.create(
-        self.args.data_path_1+"smplx_models/", 
-            model_type='smplx',
-            gender='NEUTRAL_2020', 
-            use_face_contour=False,
-            num_betas=300,
-            num_expression_coeffs=100, 
-            ext='npz',
-            use_pca=False,
-        ).to(self.rank).eval()    
-
-
 
     
     
@@ -646,8 +634,8 @@ class BaseTrainer(object):
         latent_ori = []
         l2_all = 0 
         lvel = 0
+        self.model = self.model.cuda()
         self.model.eval()
-        self.smplx.eval()
         # self.eval_copy.eval()
         with torch.no_grad():
             for its, batch_data in enumerate(self.test_loader):