diff --git a/demo/predictor.py b/demo/predictor.py
index a00735910a8371bba3e5cacc0a37846bce75ea7a..f258ce9f780e8295831310988d8e6f5bb0166109 100644
--- a/demo/predictor.py
+++ b/demo/predictor.py
@@ -110,7 +110,8 @@ class COCODemo(object):
         self.model.to(self.device)
         self.min_image_size = min_image_size
 
-        checkpointer = DetectronCheckpointer(cfg, self.model)
+        save_dir = cfg.OUTPUT_DIR
+        checkpointer = DetectronCheckpointer(cfg, self.model, save_dir=save_dir)
         _ = checkpointer.load(cfg.MODEL.WEIGHT)
 
         self.transforms = self.build_transform()