Apply suggestion from @Copilot

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: xiu-cs

scripts/FMPose3D_main.py

@@ -344,7 +344,7 @@ def print_error_action(action_error_sum, is_train):
         model_dict = model["CFM"].state_dict()
         model_path = args.model_weights_path
         print(model_path)
-        pre_dict = torch.load(model_path, weights_only=True)
+        pre_dict = torch.load(model_path, map_location=device, weights_only=True)
         for name, key in model_dict.items():
             model_dict[name] = pre_dict[name]
         model["CFM"].load_state_dict(model_dict)