allow tracing pointrend models

Reviewed By: theschnitz

Differential Revision: D25111890

fbshipit-source-id: b5412f1471e1df0555e94787b8c5798195375e1e

Author: ppwwyyxx

tools/deploy/export_model.py

@@ -13,6 +13,7 @@
 from detectron2.export import Caffe2Tracer, add_export_config
 from detectron2.export.torchscript import dump_torchscript_IR, export_torchscript_with_instances
 from detectron2.modeling import GeneralizedRCNN, build_model
+from detectron2.projects.point_rend import add_pointrend_config
 from detectron2.structures import Boxes
 from detectron2.utils.env import TORCH_VERSION
 from detectron2.utils.logger import setup_logger
@@ -23,6 +24,7 @@ def setup_cfg(args):
     # cuda context is initialized before creating dataloader, so we don't fork anymore
     cfg.DATALOADER.NUM_WORKERS = 0
     cfg = add_export_config(cfg)
+    add_pointrend_config(cfg)
     cfg.merge_from_file(args.config_file)
     cfg.merge_from_list(args.opts)
     cfg.freeze()