From 2638bdf39bc0c7d058d90e7c1a2ec5362553c0cd Mon Sep 17 00:00:00 2001
From: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Date: Thu, 5 May 2022 13:39:56 +0800
Subject: [PATCH] [Enhancement] Fix the build_segmentor in colab demo (#1551)

* [Enhancement] Delete DS_Store file

* [Enhancement] Fix the build_segmentor
---
 demo/MMSegmentation_Tutorial.ipynb | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/demo/MMSegmentation_Tutorial.ipynb b/demo/MMSegmentation_Tutorial.ipynb
index 0eb7b5a03..a75ec5c57 100644
--- a/demo/MMSegmentation_Tutorial.ipynb
+++ b/demo/MMSegmentation_Tutorial.ipynb
@@ -561,8 +561,7 @@
     "datasets = [build_dataset(cfg.data.train)]\n",
     "\n",
     "# Build the detector\n",
-    "model = build_segmentor(\n",
-    "    cfg.model, train_cfg=cfg.get('train_cfg'), test_cfg=cfg.get('test_cfg'))\n",
+    "model = build_segmentor(cfg.model)\n",
     "# Add an attribute for visualization convenience\n",
     "model.CLASSES = datasets[0].CLASSES\n",
     "\n",