Использование track.log в Ray

В Ray перед вызовом track.log необходимо выполнить track.init. Однако, если я делаю track.log вне функции, которая определяет актера, это дает мне ошибку, жалуюсь на то, что сеанс отслеживания не найден. Но если я сделаю следующее, я создам сеанс отслеживания для каждого значения гиперпарам. Правильно ли звучит нижеприведенное (мне не нравится, поскольку мы создаем один сеанс отслеживания для каждого track.log)

def train(config, reporter=None):
    track.init('track_session')
    track.log(acc=acc)

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("--redis-address")
    args = parser.parse_args()
    ray.init(redis_address=args.redis_address)
    config = {
        "l1": ray.tune.grid_search([0.001, 0.01, 0.1]),
    }
    ray.tune.run(train, config=config)

ray
person Captain Jack sparrow    schedule 04.12.2019    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Попробуй это:

def train(config):
    acc = 0
    track.log(acc=acc)

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("--redis-address")
    args = parser.parse_args()
    ray.init(redis_address=args.redis_address)
    config = {
        "l1": ray.tune.grid_search([0.001, 0.01, 0.1]),
    }
    ray.tune.run(train, config=config)
person richliaw    schedule 09.12.2019