Skip to content

Commit 2638bdf

Browse files
authored
[Enhancement] Fix the build_segmentor in colab demo (open-mmlab#1551)
* [Enhancement] Delete DS_Store file * [Enhancement] Fix the build_segmentor
1 parent 2d8be37 commit 2638bdf

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

demo/MMSegmentation_Tutorial.ipynb

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -561,8 +561,7 @@
561561
"datasets = [build_dataset(cfg.data.train)]\n",
562562
"\n",
563563
"# Build the detector\n",
564-
"model = build_segmentor(\n",
565-
" cfg.model, train_cfg=cfg.get('train_cfg'), test_cfg=cfg.get('test_cfg'))\n",
564+
"model = build_segmentor(cfg.model)\n",
566565
"# Add an attribute for visualization convenience\n",
567566
"model.CLASSES = datasets[0].CLASSES\n",
568567
"\n",

0 commit comments

Comments
 (0)