add naswot

graph_dit/naswot/models/cell_searchs/_test_module.py

@@ -0,0 +1,12 @@
+##################################################
+# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
+##################################################
+import torch
+from search_model_enas_utils import Controller
+
+def main():
+  controller = Controller(6, 4)
+  predictions = controller()
+
+if __name__ == '__main__':
+  main()