Spaces:
Runtime error
Runtime error
Update src/sample_utils/pose_sampler.py
Browse files
src/sample_utils/pose_sampler.py
CHANGED
|
@@ -20,9 +20,9 @@ class PoseSampler:
|
|
| 20 |
在此处加载配置、统计量和模型,确保这些重型操作只执行一次。
|
| 21 |
"""
|
| 22 |
self.device = torch.device(device)
|
| 23 |
-
self.checkpoint_path = "
|
| 24 |
self.dataset = dataset
|
| 25 |
-
self.stats_dir = "
|
| 26 |
|
| 27 |
# 1. 自动检测 run_dir 并加载配置
|
| 28 |
self.run_dir = self._detect_run_dir(self.checkpoint_path)
|
|
|
|
| 20 |
在此处加载配置、统计量和模型,确保这些重型操作只执行一次。
|
| 21 |
"""
|
| 22 |
self.device = torch.device(device)
|
| 23 |
+
self.checkpoint_path = "src/sample_utils/lib/ckpt/fm_20251117_172225/checkpoints/best.pt"
|
| 24 |
self.dataset = dataset
|
| 25 |
+
self.stats_dir = "src/sample_utils/lib/data_stats"
|
| 26 |
|
| 27 |
# 1. 自动检测 run_dir 并加载配置
|
| 28 |
self.run_dir = self._detect_run_dir(self.checkpoint_path)
|