Update NATS-Bench (sss version 1.0)

2020-08-28 06:02:35 +00:00
parent 3529b993ff
commit c68458f66c
9 changed files with 413 additions and 28 deletions
--- a/lib/utils/init.py
+++ b/lib/utils/init.py
@@ -3,3 +3,4 @@ from .gpu_manager      import GPUManager
 from .flop_benchmark   import get_model_infos, 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