Update NATS-Bench (sss version 1.0)

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

lib/utils/hash_utils.py

@@ -0,0 +1,16 @@
+import os, hashlib
+
+
+def get_md5_file(file_path, post_truncated=5):
+  md5_hash = hashlib.md5()
+  if os.path.exists(file_path):
+    xfile = open(file_path, "rb")
+    content = xfile.read()
+    md5_hash.update(content)
+    digest = md5_hash.hexdigest()
+  else:
+    raise ValueError('[get_md5_file] {:} does not exist'.format(file_path))
+  if post_truncated is None:
+    return digest
+  else:
+    return digest[-post_truncated:]