Куда импортировать тензорный поток с использованием лучевых акторов? - PullRequest
Новая
       34

Куда импортировать тензорный поток с использованием лучевых акторов?

0 голосов
/ 26 января 2020

Используя лучевые акторы для параллельного запуска нескольких моделей тензорного потока, я спрашиваю себя, куда импортировать тензорный поток: 

# [1] maybe import tensorflow here?

@ray.remote(num_cpus=1)
class Remote_Runner:
    # [2] maybe import tensorflow here?
    def __init__(self, weights):
        # [3] maybe import tensorflow here?
        self.model=My_model()
        self.model.set_weights(wegihts)

    def do_something_with_model:
        self.model.do_something()

В примере, приведенном в документации, упоминается импорт тензорного потока в акторе из-за 'побочных эффектов импорт TensorFlow и установка глобального состояния ', но только приводит пример удаленной функции луча. Так я должен запустить 'импортировать тензор потока как tf' в [1], [2] или [3], или даже где-нибудь еще? Есть ли передовая практика, которой следует следовать, каковы различия между [1], [2] и [3], т.е. как я могу получить доступ к тензорному потоку в каждом случае и когда они выполняются?

1 Ответ

1 голос
/ 31 января 2020

Возможно, это сейчас исправлено, но самый безопасный вариант для рассмотрения - это импортирование тензорного потока в [3].

...