From 15071c02f6225352c303cd36ca037f77ef2fcdf6 Mon Sep 17 00:00:00 2001 From: unknown <1874859902@qq.com> Date: Fri, 24 Sep 2021 18:10:02 +0800 Subject: [PATCH] val_step add optimizer --- mmseg/models/segmentors/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/models/segmentors/base.py b/mmseg/models/segmentors/base.py index 906c6fe564..8ccf574564 100644 --- a/mmseg/models/segmentors/base.py +++ b/mmseg/models/segmentors/base.py @@ -145,7 +145,7 @@ def train_step(self, data_batch, optimizer, **kwargs): return outputs - def val_step(self, data_batch, **kwargs): + def val_step(self, data_batch,optimizer=None, **kwargs): """The iteration step during validation. This method shares the same signature as :func:`train_step`, but used