fix the bug in inferen.py

inference.py

@@ -87,10 +87,10 @@ class Inference(object):
 
         model_path = args.path
         checkpoint = torch.load(model_path.replace("best.pt", "best_part_1.pt"))
-        checkpoint["model"].update(model_path.replace("best.pt", "best_part_2.pt"))
-        checkpoint["model"].update(model_path.replace("best.pt", "best_part_3.pt"))
+        checkpoint["model"].update(torch.load(model_path.replace("best.pt", "best_part_2.pt")))
+        checkpoint["model"].update(torch.load(model_path.replace("best.pt", "best_part_3.pt")))
         torch.save(checkpoint, model_path)
-        # load part 1
+
         state = torch.load(args.path, map_location=torch.device("cpu"))
         cfg_args = eval(str(state["cfg"]))["model"]
         del cfg_args["_name"]
@@ -178,4 +178,4 @@ class Inference(object):
                     score = hypo['score'] / math.log(2)  # convert to base 2
                     tmp_res.append([detok_hypo_str, score])
             final_results.append(tmp_res)
-        return final_results
+        return final_results