Update NATS-Bench (tss version 0.8)

2020-08-28 08:31:53 +00:00
parent c68458f66c
commit 2c86d6aa67
5 changed files with 172 additions and 111 deletions
--- a/lib/utils/init.py
+++ b/lib/utils/init.py
@@ -4,3 +4,4 @@ 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
+from .str_utils        import split_str2indexes