update ENAS
This commit is contained in:
9
lib/models/cell_searchs/_test_module.py
Normal file
9
lib/models/cell_searchs/_test_module.py
Normal file
@@ -0,0 +1,9 @@
|
||||
import torch
|
||||
from search_model_enas_utils import Controller
|
||||
|
||||
def main():
|
||||
controller = Controller(6, 4)
|
||||
predictions = controller()
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
Reference in New Issue
Block a user