##################################################
# 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()