atnikos commited on
Commit
d425bee
·
1 Parent(s): 08db8da

fix gpu issue

Browse files
Files changed (1) hide show
  1. normalization.py +1 -1
normalization.py CHANGED
@@ -14,7 +14,7 @@ class Normalizer:
14
  "body_joints_local_wo_z_rot"],
15
  dim_per_feat: List[int] = [3, 6, 6, 126, 66], *args, **kwargs):
16
 
17
- self.stats = self.load_norm_statistics(statistics_path, 'cuda')
18
  # from src.model.utils.tools import pack_to_render
19
  # mr = pack_to_render(aa.detach().cpu(), trans=None)
20
  # mr = {k: v[0] for k, v in mr.items()}
 
14
  "body_joints_local_wo_z_rot"],
15
  dim_per_feat: List[int] = [3, 6, 6, 126, 66], *args, **kwargs):
16
 
17
+ self.stats = self.load_norm_statistics(statistics_path, 'cpu')
18
  # from src.model.utils.tools import pack_to_render
19
  # mr = pack_to_render(aa.detach().cpu(), trans=None)
20
  # mr = {k: v[0] for k, v in mr.items()}