"docs/source/vscode:/vscode.git/clone" did not exist on "fe2b6ca6e8cdf652e36d48f5a88c58f13c53ad8c"
Commit f82d44d3 authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

update runner in d2go_beginner script

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/76

Detectron2GoRunner doesn't contain configs about exporting RCNN models, use GeneralizedRCNNRunner instead

Reviewed By: zhanghang1989

Differential Revision: D28652627

fbshipit-source-id: 6f324f608d8b2abdf98179a36e4b79837f135340
parent 57809b0f
......@@ -421,11 +421,11 @@
}
],
"source": [
"from d2go.runner import Detectron2GoRunner\n",
"from d2go.runner import GeneralizedRCNNRunner\n",
"\n",
"\n",
"def prepare_for_launch():\n",
" runner = Detectron2GoRunner()\n",
" runner = GeneralizedRCNNRunner()\n",
" cfg = runner.get_default_cfg()\n",
" cfg.merge_from_file(model_zoo.get_config_file(\"faster_rcnn_fbnetv3a_C4.yaml\"))\n",
" cfg.MODEL_EMA.ENABLED = False\n",
......@@ -709,4 +709,4 @@
},
"nbformat": 4,
"nbformat_minor": 4
}
\ No newline at end of file
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment