xautodl/lib/utils/__init__.py
2019-09-28 18:24:47 +10:00

7 lines
286 B
Python

##################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
##################################################
from .evaluation_utils import obtain_accuracy
from .gpu_manager import GPUManager
from .flop_benchmark import get_model_infos