Update dataloaders/beat_sep_single.py

dataloaders/beat_sep_single.py

@@ -76,16 +76,15 @@ class CustomDataset(Dataset):
                 else:
                     self.joint_mask[self.ori_joint_list[joint_name][1] - self.ori_joint_list[joint_name][0]:self.ori_joint_list[joint_name][1]] = 1
         # select trainable joints
-        self.smplx = smplx.create(
-            self.args.data_path_1+"smplx_models/", 
-            model_type='smplx',
-            gender='NEUTRAL_2020', 
+        self.smplx = smplx.SMPLX(
+            model_path=self.args.data_path_1 + "smplx_models/",
+            gender='NEUTRAL_2020',
             use_face_contour=False,
             num_betas=300,
-            num_expression_coeffs=100, 
+            num_expression_coeffs=100,
             ext='npz',
             use_pca=False,
-            ).eval()
+        ).eval()
 
         if loader_type == 'test':
             # In demo/test mode, skip dataset CSV and use provided paths