From 1d677500bf838fe4623d0d2a72ef8e78e12c8e4b Mon Sep 17 00:00:00 2001
From: Ilija Radosavovic <ilijar@fb.com>
Date: Tue, 3 Apr 2018 08:34:56 -0700
Subject: [PATCH] Specify weights when loading the model in the pkl to pb
 conversion tool

Reviewed By: rbgirshick

Differential Revision: D7484464

fbshipit-source-id: f92804fedc3ae040e585e7c0c2eb3e44526eef22
---
 tools/convert_pkl_to_pb.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/convert_pkl_to_pb.py b/tools/convert_pkl_to_pb.py
index bc61938..ae2b94e 100644
--- a/tools/convert_pkl_to_pb.py
+++ b/tools/convert_pkl_to_pb.py
@@ -333,7 +333,7 @@ def _save_models(all_net, all_init_net, args):
 
 
 def load_model(args):
-    model = test_engine.initialize_model_from_cfg()
+    model = test_engine.initialize_model_from_cfg(cfg.TEST.WEIGHTS)
     blobs = mutils.get_ws_blobs()
 
     return model, blobs