fix(actpred): gen_calibration --deploy 的 is_range_mode 未导入即使用
import 在 if 分支内滞后于首次调用,fixed_range 归一化模型生成校准集 时直接 NameError;导入提前到 build_deploy_set 开头。
This commit is contained in:
1 parent
d8b601ab9a
commit
293d71e86b
1 file changed
+1
@@ -86,6 +86,7 @@ def build_image_samples(n_samples, frame_len, height, width, rng,
|
||||
|
||||
def build_deploy_set(args):
|
||||
import torch
|
||||
from action_norm import is_range_mode, normalize_action
|
||||
from model import ActionPredictor
|
||||
|
||||
ckpt = torch.load(args.model, map_location="cpu", weights_only=True)
|
||||
|
||||
Reference in new issue
Block a user