fix GuyTevet#83: Fix bug related to action-to-motion.

zhongshijun · Feb 15, 2023 · fd39c9e · fd39c9e
1 parent c283984
commit fd39c9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eval/a2m/gru_eval.py b/eval/a2m/gru_eval.py
@@ -40,7 +40,7 @@ def __init__(self, mode, model, diffusion, dataiterator, device, unconstrained,
                                                     translation=True, jointstype='smpl', vertstrans=True, betas=None,
                                                     beta=0, glob_rot=None, get_rotations_back=False)
                 batch["lengths"] = model_kwargs['y']['lengths'].to(device)
-                if unconstrained:  # proceed only if not running unconstrained
+                if not unconstrained:  # proceed only if not running unconstrained
                     batch["y"] = model_kwargs['y']['action'].squeeze().long().cpu()  # using torch.long so lengths/action will be used as indices
                 self.batches.append(batch)