To re-org Q-results

2021-03-06 06:13:22 -08:00
parent cc28e1589e
commit 511e2443d0
7 changed files with 172 additions and 40 deletions
--- a/lib/utils/init.py
+++ b/lib/utils/init.py
@@ -1,6 +1,6 @@
 from .evaluation_utils import obtain_accuracy
 from .gpu_manager      import GPUManager
-from .flop_benchmark   import get_model_infos, count_parameters_in_MB
+from .flop_benchmark   import get_model_infos, count_parameters, count_parameters_in_MB
 from .affine_utils     import normalize_points, denormalize_points
 from .affine_utils     import identity2affine, solve2theta, affine2image
 from .hash_utils       import get_md5_file