[Fix] Revert sync bn in inferencer (#1832)

pull/1840/head
Tong Gao 2023-04-04 10:14:51 +08:00 committed by GitHub
parent e6174b29fe
commit 16de16f8f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -7,6 +7,7 @@ import mmengine
import numpy as np
from mmengine.dataset import Compose
from mmengine.infer.infer import BaseInferencer, ModelType
from mmengine.model.utils import revert_sync_batchnorm
from mmengine.registry import init_default_scope
from mmengine.structures import InstanceData
from rich.progress import track
@ -63,6 +64,7 @@ class BaseMMOCRInferencer(BaseInferencer):
init_default_scope(scope)
super().__init__(
model=model, weights=weights, device=device, scope=scope)
self.model = revert_sync_batchnorm(self.model)
def preprocess(self, inputs: InputsType, batch_size: int = 1, **kwargs):
"""Process the inputs into a model-feedable format.