Update data_loaders/humanml/data/dataset.py

data_loaders/humanml/data/dataset.py

@@ -786,14 +786,8 @@ class HumanML3D(data.Dataset):
             self.std = np.load(pjoin(opt.meta_dir, f'{opt.dataset_name}_std.npy'))
         elif mode in ['train', 'eval', 'text_only']:
             # used by our models
-            if opt.dataset_name == 't2m':
-                self.mean = np.load(pjoin(opt.meta_dir, 't2m_mean.npy'))
-                self.std = np.load(pjoin(opt.meta_dir, 't2m_std.npy'))
-            elif opt.dataset_name == 'kit':
-                self.mean = np.load(pjoin(opt.meta_dir, 'kit_mean.npy'))
-                self.std = np.load(pjoin(opt.meta_dir, 'kit_std.npy'))
-            else:
-                raise ValueError(f'Unknown dataset_name: {opt.dataset_name}')
+            self.mean = np.load(pjoin(opt.data_root, 'Mean.npy'))
+            self.std = np.load(pjoin(opt.data_root, 'Std.npy'))
 
         if mode == 'eval':
             # used by T2M models (including evaluators)