Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
nas.py 1.52 KB
Newer Older
Zejun Lin's avatar
Unify nas searchspace (#1393)  
Zejun Lin committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49 
import nni
import time


def add_one(inputs):
    return inputs + 1


def add_two(inputs):
    return inputs + 2


def add_three(inputs):
    return inputs + 3


def add_four(inputs):
    return inputs + 4


def main():
    images = 5
    layer_1_out = nni.mutable_layer('mutable_block_39', 'mutable_layer_0',
        {'add_one()': add_one, 'add_two()': add_two, 'add_three()':
        add_three, 'add_four()': add_four}, {'add_one()': {}, 'add_two()':
        {}, 'add_three()': {}, 'add_four()': {}}, [], {'images': images}, 1,
        'classic_mode')
    layer_2_out = nni.mutable_layer('mutable_block_39', 'mutable_layer_1',
        {'add_one()': add_one, 'add_two()': add_two, 'add_three()':
        add_three, 'add_four()': add_four}, {'add_one()': {}, 'add_two()':
        {}, 'add_three()': {}, 'add_four()': {}}, [], {'layer_1_out':
        layer_1_out}, 1, 'classic_mode')
    layer_3_out = nni.mutable_layer('mutable_block_39', 'mutable_layer_2',
        {'add_one()': add_one, 'add_two()': add_two, 'add_three()':
        add_three, 'add_four()': add_four}, {'add_one()': {}, 'add_two()':
        {}, 'add_three()': {}, 'add_four()': {}}, [], {'layer_1_out':
        layer_1_out, 'layer_2_out': layer_2_out}, 1, 'classic_mode')
    nni.report_intermediate_result(layer_1_out)
    time.sleep(2)
    nni.report_intermediate_result(layer_2_out)
    time.sleep(2)
    nni.report_intermediate_result(layer_3_out)
    time.sleep(2)
    layer_3_out = layer_3_out + 10
    nni.report_final_result(layer_3_out)


if __name__ == '__main__':
    main()