2019-11-15 07:15:07 +01:00
|
|
|
##################################################
|
|
|
|
|
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2019 #
|
|
|
|
|
##################################################
|
2019-11-08 15:36:31 +01:00
|
|
|
import torch
|
|
|
|
|
from search_model_enas_utils import Controller
|
|
|
|
|
|
2021-05-12 10:28:05 +02:00
|
|
|
|
2019-11-08 15:36:31 +01:00
|
|
|
def main():
|
2021-05-12 10:28:05 +02:00
|
|
|
controller = Controller(6, 4)
|
|
|
|
|
predictions = controller()
|
|
|
|
|
|
2019-11-08 15:36:31 +01:00
|
|
|
|
2021-05-12 10:28:05 +02:00
|
|
|
if __name__ == "__main__":
|
|
|
|
|
main()
|
