autodl-projects/lib/models/l2s_cell_searchs/_test_module.py

13 lines
324 B
Python
Raw Normal View History

2019-11-19 01:58:04 +01:00
##################################################
# 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()