Skip to content

Commit d118793

Browse files
committed
remove common
Signed-off-by: changwa1 <[email protected]>
1 parent c4e3520 commit d118793

File tree

25 files changed

+2472
-3210
lines changed

25 files changed

+2472
-3210
lines changed

neural_compressor/adaptor/mxnet_utils/util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
from enum import Enum
2525
from tempfile import TemporaryDirectory
2626
from neural_compressor.utils.utility import LazyImport
27-
from neural_compressor.model.model import MXNetModel as NCModel
27+
from neural_compressor.model.mxnet_model import MXNetModel as NCModel
2828

2929
mx = LazyImport("mxnet")
3030

neural_compressor/adaptor/tensorflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def train(self, model, dataloader, optimizer_tuple,
133133
criterion_tuple, hooks, postprocess, **kwargs):
134134
# check model is savedmodel or not
135135
import tensorflow as tf
136-
from neural_compressor.model.model import get_model_type
136+
from neural_compressor.model.tensorflow_model import get_model_type
137137
tf.random.set_seed(1)
138138
self.model_type = get_model_type(model._model)
139139
optimizer = optimizer_tuple[0](**optimizer_tuple[1])
@@ -1212,7 +1212,7 @@ def inspect_tensor(self, model, dataloader=None, op_list=[], iteration_list=[],
12121212
]
12131213
}
12141214
"""
1215-
from neural_compressor.model.model import TensorflowBaseModel
1215+
from neural_compressor.model.tensorflow_model import TensorflowBaseModel
12161216
from neural_compressor.utils.utility import load_data_from_pkl, dump_data_to_local
12171217
from neural_compressor.adaptor.tf_utils.graph_util import GraphAnalyzer
12181218
from .tf_utils.util import int8_node_name_reverse

neural_compressor/common/__init__.py

Lines changed: 0 additions & 28 deletions
This file was deleted.

0 commit comments

Comments
 (0)