diff --git a/torchbenchmark/operator_loader/__init__.py b/torchbenchmark/operator_loader/__init__.py index 7ef561459..2fef81efa 100644 --- a/torchbenchmark/operator_loader/__init__.py +++ b/torchbenchmark/operator_loader/__init__.py @@ -14,7 +14,8 @@ BenchmarkOperator, register_benchmark_mannually, ) -from .operator_inp_utils import aten, OperatorInputsLoader, to_channels_last # noqa + +from .operator_inp_utils import aten, OperatorInputsLoader, to_channels_last timm_loader = OperatorInputsLoader.get_timm_loader() huggingface_loader = OperatorInputsLoader.get_huggingface_loader()