Fixed issue #42

GuyTevet · Nov 15, 2022 · 34c8a51 · 34c8a51
1 parent fc1439f
commit 34c8a51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eval/eval_humanml.py b/eval/eval_humanml.py
@@ -279,7 +279,7 @@ def evaluation(eval_wrapper, gt_loader, eval_motion_loaders, log_file, replicati
     num_actions = gen_loader.dataset.num_actions
 
     logger.log("Creating model and diffusion...")
-    model, diffusion = create_model_and_diffusion(args)
+    model, diffusion = create_model_and_diffusion(args, gen_loader)
 
     logger.log(f"Loading checkpoints from [{args.model_path}]...")
     state_dict = torch.load(args.model_path, map_location='cpu')