Move to xautodl
This commit is contained in:
14
xautodl/models/cell_searchs/_test_module.py
Normal file
14
xautodl/models/cell_searchs/_test_module.py
Normal file
@@ -0,0 +1,14 @@
|
||||
##################################################
|
||||
# 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()
|
Reference in New Issue
Block a user