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

@@ -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