yolozyk commited on
Commit
ddda209
·
verified ·
1 Parent(s): 695f42e

Update src/sample_utils/pose_sampler.py

Browse files
Files changed (1) hide show
  1. src/sample_utils/pose_sampler.py +2 -2
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 = "/app/src/sample_utils/lib/ckpt/fm_20251117_172225/checkpoints/best.pt"
24
  self.dataset = dataset
25
- self.stats_dir = "/app/src/sample_utils/lib/data_stats"
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)