Prototype generic nas model.

2020-07-16 10:34:34 +00:00
parent a99df6dc31
commit 68f9d037eb
7 changed files with 510 additions and 4 deletions
--- a/lib/utils/init.py
+++ b/lib/utils/init.py
@@ -1,5 +1,5 @@
 from .evaluation_utils import obtain_accuracy
 from .gpu_manager      import GPUManager
-from .flop_benchmark   import get_model_infos
+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