from .evaluation_utils import obtain_accuracy from .gpu_manager import GPUManager from .flop_benchmark import get_model_infos