Skip to content

Commit 6ee190e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f209286 commit 6ee190e

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

monai/handlers/model_calibrator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__(
4545
model: torch.nn.Module,
4646
export_path: str,
4747
config: dict= mtq.INT8_SMOOTHQUANT_CFG,
48-
48+
4949
) -> None:
5050
self.model = model
5151
self.export_path = export_path
@@ -57,7 +57,7 @@ def attach(self, engine: Engine) -> None:
5757
engine: Ignite Engine, it can be a trainer, validator or evaluator.
5858
"""
5959
engine.add_event_handler(Events.STARTED, self)
60-
60+
6161
@staticmethod
6262
def _model_wrapper(engine, model):
6363
engine.run()

monai/handlers/model_quantizer.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,14 +11,12 @@
1111

1212
from __future__ import annotations
1313

14-
import logging
15-
import warnings
1614
from types import MethodType
17-
from typing import TYPE_CHECKING, Sequence
15+
from typing import TYPE_CHECKING
16+
from collections.abc import Sequence
1817

1918
import torch
2019

21-
from monai.networks.utils import copy_model_state
2220
from monai.utils import IgniteInfo, min_version, optional_import
2321
from torch.ao.quantization.quantizer import Quantizer
2422
from torch.ao.quantization.quantizer.xnnpack_quantizer import (
@@ -27,7 +25,6 @@
2725
)
2826
from torch.ao.quantization.quantize_pt2e import (
2927
prepare_qat_pt2e,
30-
convert_pt2e,
3128
)
3229

3330
Events, _ = optional_import("ignite.engine", IgniteInfo.OPT_IMPORT_VERSION, min_version, "Events")
@@ -57,7 +54,7 @@ def __init__(
5754
example_inputs: Sequence,
5855
export_path: str,
5956
quantizer: Quantizer | None = None,
60-
57+
6158
) -> None:
6259
self.model = model
6360
self.example_inputs = example_inputs
@@ -77,6 +74,6 @@ def start(self) -> None:
7774
self.model = prepare_qat_pt2e(self.model, self.quantizer)
7875
self.model.train = MethodType(torch.ao.quantization.move_exported_model_to_train, self.model)
7976
self.model.eval = MethodType(torch.ao.quantization.move_exported_model_to_eval, self.model)
80-
77+
8178
def epoch(self) -> None:
8279
torch.save(self.model.state_dict(), self.export_path)

0 commit comments

Comments
 (0)