10 lines
171 B
Python
10 lines
171 B
Python
|
import torch
|
||
|
|
from search_model_enas_utils import Controller
|
||
|
|
|
||
|
|
def main():
|
||
|
|
controller = Controller(6, 4)
|
||
|
|
predictions = controller()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|