Spaces:
Running
Running
fix gpu issue
Browse files- 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, '
|
| 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()}
|