Skip to content

Commit 2414e5d

Browse files
ClarkChin08ftian1
authored andcommitted
[fix] fix the import lpot.data to lpot.experimental
1 parent 9c9324d commit 2414e5d

File tree

5 files changed

+10
-10
lines changed

5 files changed

+10
-10
lines changed

examples/onnxrt/language_translation/bert/bert_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -264,8 +264,8 @@ def eval_func(model):
264264
if args.benchmark:
265265
model = onnx.load(args.model_path)
266266

267-
from lpot.data.datasets.dummy_dataset import DummyDataset
268-
from lpot.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
267+
from lpot.experimental.data.datasets.dummy_dataset import DummyDataset
268+
from lpot.experimental.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
269269
shapes, lows, highs = parse_dummy_input(model, args.benchmark_nums, args.max_seq_length)
270270
dummy_dataset = DummyDataset(shapes, low=lows, high=highs, dtype="int64")
271271
dummy_dataloader = ONNXRTDataLoader(dummy_dataset)

examples/onnxrt/language_translation/distilbert/bert_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -269,8 +269,8 @@ def eval_func(model):
269269
if args.benchmark:
270270
model = onnx.load(args.model_path)
271271

272-
from lpot.data.datasets.dummy_dataset import DummyDataset
273-
from lpot.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
272+
from lpot.experimental.data.datasets.dummy_dataset import DummyDataset
273+
from lpot.experimental.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
274274
shapes, lows, highs = parse_dummy_input(model, args.benchmark_nums, args.max_seq_length)
275275
dummy_dataset = DummyDataset(shapes, low=lows, high=highs, dtype="int64")
276276
dummy_dataloader = ONNXRTDataLoader(dummy_dataset)

examples/onnxrt/language_translation/mobilebert/bert_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -265,8 +265,8 @@ def eval_func(model):
265265
if args.benchmark:
266266
model = onnx.load(args.model_path)
267267

268-
from lpot.data.datasets.dummy_dataset import DummyDataset
269-
from lpot.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
268+
from lpot.experimental.data.datasets.dummy_dataset import DummyDataset
269+
from lpot.experimental.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
270270
shapes, lows, highs = parse_dummy_input(model, args.benchmark_nums, args.max_seq_length)
271271
dummy_dataset = DummyDataset(shapes, low=lows, high=highs, dtype="int64")
272272
dummy_dataloader = ONNXRTDataLoader(dummy_dataset)

examples/onnxrt/language_translation/roberta/bert_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -282,8 +282,8 @@ def eval_func(model):
282282
if args.benchmark:
283283
model = onnx.load(args.model_path)
284284

285-
from lpot.data.datasets.dummy_dataset import DummyDataset
286-
from lpot.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
285+
from lpot.experimental.data.datasets.dummy_dataset import DummyDataset
286+
from lpot.experimental.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
287287
shapes, lows, highs = parse_dummy_input(model, args.benchmark_nums, args.max_seq_length)
288288
dummy_dataset = DummyDataset(shapes, low=lows, high=highs, dtype="int64")
289289
dummy_dataloader = ONNXRTDataLoader(dummy_dataset)

examples/onnxrt/oob_general.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,8 @@ def parse_dummy_input(model, benchmark_nums):
100100
input_shapes = [shape.split('x') for shape in input_shapes]
101101
shapes = [tuple([args.benchmark_nums] + [int(dim) for dim in shape]) for shape in input_shapes]
102102

103-
from lpot.data.datasets.dummy_dataset import DummyDataset
104-
from lpot.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
103+
from lpot.experimental.data.datasets.dummy_dataset import DummyDataset
104+
from lpot.experimental.data.dataloaders.onnxrt_dataloader import ONNXRTDataLoader
105105
dummy_dataset = DummyDataset(shapes, low=lows, high=highs, dtype=dtypes, label=True)
106106
dummy_dataloader = ONNXRTDataLoader(dummy_dataset, batch_size=args.eval_batch_size)
107107

0 commit comments

Comments
 (0)