From eeba084341aaa454ce13cb32fa7fd9282fc73a67 Mon Sep 17 00:00:00 2001
From: Shiyu <ligary04@hotmail.com>
Date: Mon, 13 Nov 2023 13:52:37 +0800
Subject: [PATCH] [fix] replace ema_model with model in demo/test_ap_on_coco
 (#242)

---
 demo/test_ap_on_coco.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/test_ap_on_coco.py b/demo/test_ap_on_coco.py
index 59ce6a2..f1e532f 100644
--- a/demo/test_ap_on_coco.py
+++ b/demo/test_ap_on_coco.py
@@ -26,7 +26,7 @@ def load_model(model_config_path: str, model_checkpoint_path: str, device: str =
     args.device = device
     model = build_model(args)
     checkpoint = torch.load(model_checkpoint_path, map_location="cpu")
-    model.load_state_dict(clean_state_dict(checkpoint["ema_model"]), strict=False)
+    model.load_state_dict(clean_state_dict(checkpoint["model"]), strict=False)
     model.eval()
     return model