raise exception if model_cfg[N_FRAMES_PER_CLIP] is not a factor of nu…

…mber of frames
talmolab · Sep 5, 2024 · 069d5bd · 069d5bd
1 parent 0b5ce97
commit 069d5bd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/stac_mjx/main.py b/stac_mjx/main.py
@@ -72,7 +72,9 @@ def run_stac(
     if stac_cfg.skip_transform == 1:
         logging.info("skipping transform()")
         return fit_path, None
-
+    elif kp_data.shape[0] % model_cfg["N_FRAMES_PER_CLIP"] != 0:
+        raise ValueError(f"N_FRAMES_PER_CLIP ({model_cfg["N_FRAMES_PER_CLIP"]}) must be a factor of the number of mocap frames ({kp_data.shape[0]})")
+
     logging.info("Running transform()")
     with open(fit_path, "rb") as file:
         fit_data = pickle.load(file)