From c498e6f593adc9b7c502f22d87b2d7d85a9bcb67 Mon Sep 17 00:00:00 2001
From: Miao Zheng <76149310+MeowZheng@users.noreply.github.com>
Date: Tue, 9 Aug 2022 23:33:04 +0800
Subject: [PATCH] [Fix] Add SegVisualizationHook in default hooks (#1900)

---
 configs/_base_/schedules/schedule_160k.py | 2 +-
 configs/_base_/schedules/schedule_20k.py  | 2 +-
 configs/_base_/schedules/schedule_320k.py | 2 +-
 configs/_base_/schedules/schedule_40k.py  | 2 +-
 configs/_base_/schedules/schedule_80k.py  | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/configs/_base_/schedules/schedule_160k.py b/configs/_base_/schedules/schedule_160k.py
index 1055958a3..6abc80250 100644
--- a/configs/_base_/schedules/schedule_160k.py
+++ b/configs/_base_/schedules/schedule_160k.py
@@ -22,4 +22,4 @@ default_hooks = dict(
     param_scheduler=dict(type='ParamSchedulerHook'),
     checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=16000),
     sampler_seed=dict(type='DistSamplerSeedHook'),
-)
+    visualization=dict(type='SegVisualizationHook'))
diff --git a/configs/_base_/schedules/schedule_20k.py b/configs/_base_/schedules/schedule_20k.py
index b4bc083de..6a64ca1b9 100644
--- a/configs/_base_/schedules/schedule_20k.py
+++ b/configs/_base_/schedules/schedule_20k.py
@@ -21,4 +21,4 @@ default_hooks = dict(
     param_scheduler=dict(type='ParamSchedulerHook'),
     checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=2000),
     sampler_seed=dict(type='DistSamplerSeedHook'),
-)
+    visualization=dict(type='SegVisualizationHook'))
diff --git a/configs/_base_/schedules/schedule_320k.py b/configs/_base_/schedules/schedule_320k.py
index 1c6f3500e..902dff843 100644
--- a/configs/_base_/schedules/schedule_320k.py
+++ b/configs/_base_/schedules/schedule_320k.py
@@ -22,4 +22,4 @@ default_hooks = dict(
     param_scheduler=dict(type='ParamSchedulerHook'),
     checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=32000),
     sampler_seed=dict(type='DistSamplerSeedHook'),
-)
+    visualization=dict(type='SegVisualizationHook'))
diff --git a/configs/_base_/schedules/schedule_40k.py b/configs/_base_/schedules/schedule_40k.py
index d768012e8..e682aeaa2 100644
--- a/configs/_base_/schedules/schedule_40k.py
+++ b/configs/_base_/schedules/schedule_40k.py
@@ -21,4 +21,4 @@ default_hooks = dict(
     param_scheduler=dict(type='ParamSchedulerHook'),
     checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=4000),
     sampler_seed=dict(type='DistSamplerSeedHook'),
-)
+    visualization=dict(type='SegVisualizationHook'))
diff --git a/configs/_base_/schedules/schedule_80k.py b/configs/_base_/schedules/schedule_80k.py
index 170a2ecb8..bb6b044ab 100644
--- a/configs/_base_/schedules/schedule_80k.py
+++ b/configs/_base_/schedules/schedule_80k.py
@@ -21,4 +21,4 @@ default_hooks = dict(
     param_scheduler=dict(type='ParamSchedulerHook'),
     checkpoint=dict(type='CheckpointHook', by_epoch=False, interval=8000),
     sampler_seed=dict(type='DistSamplerSeedHook'),
-)
+    visualization=dict(type='SegVisualizationHook'))