"test/ut/nas/test_graph.py" did not exist on "45e82b3e5a30c671fbee3462c2e655c916c3287e"
naive_tuner.py 1.06 KB
Newer Older
Deshui Yu's avatar
Deshui Yu committed
1
2
3
4
5
6
7
import json
import logging

from nni.tuner import Tuner

_logger = logging.getLogger('NaiveTuner')
_logger.info('start')
8
_result = open('/tmp/nni_tuner_result.txt', 'w')
Deshui Yu's avatar
Deshui Yu committed
9
10

class NaiveTuner(Tuner):
11
    def __init__(self, optimize_mode):
Deshui Yu's avatar
Deshui Yu committed
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
        self.cur = 0
        _logger.info('init')

    def generate_parameters(self, parameter_id):
        self.cur += 1
        _logger.info('generate parameters: %s' % self.cur)
        return { 'x': self.cur }

    def receive_trial_result(self, parameter_id, parameters, reward):
        _logger.info('receive trial result: %s, %s, %s' % (parameter_id, parameters, reward))
        _result.write('%d %d\n' % (parameters['x'], reward))
        _result.flush()

    def update_search_space(self, search_space):
        _logger.info('update_search_space: %s' % search_space)
27
        with open('/tmp/nni_tuner_search_space.json', 'w') as file_:
Deshui Yu's avatar
Deshui Yu committed
28
29
            json.dump(search_space, file_)

30
31
32
33
34
35
36
    def _on_exit(self):
        _result.write('DONE\n')
        _result.close()

    def _on_error(self):
        _result.write('ERROR\n')
        _result.close()