Tensorflow Tensoroard (Ngrok) в Google Colab - PullRequest
Новая
       86

Tensorflow Tensoroard (Ngrok) в Google Colab

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

Я пытаюсь создать собственную модель обнаружения объектов в Google Colab. У меня проблема с запуском Tensorboard. Непосредственно перед началом обучения я запускаю следующие команды для Ngrok. 

!wget https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip
!unzip ngrok-stable-linux-amd64.zip

LOG_DIR = '/tmp/log'
get_ipython().system_raw(
    'tensorboard --logdir {} --host 0.0.0.0 --port 6006 &'
    .format(LOG_DIR)
)

get_ipython().system_raw('./ngrok http 6006 &')

!curl -s http://localhost:4040/api/tunnels | python3 -c \
    "import sys, json; print(json.load(sys.stdin)['tunnels'][0]['public_url'])"

И получил вывод с URL-адресом вроде this .

Затем я приступил к обучению и после этого перешел на URL-адрес. Но данных там не найдено. Может кто-нибудь посоветовать, где я ошибаюсь.

Также еще один квест ... Я нашел, что для остановки обучения мне нужно использовать Ctrl + C. Но как отправить Ctrl + C на ноутбуке Colab?

Заранее спасибо.

С уважением.

1 Ответ

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

Использование Ngrok для получения туннеля к экземпляру TensorBoard в Colab раньше было довольно распространенной практикой, но больше не нужно: TensorBoard имеет встроенную поддержку для сред ноутбуков, включая Colab и Jupyter. После загрузки расширения для ноутбука tensorboard вы можете просто запустить командную строку с %tensorboard, и она должна просто работать: 

%load_ext tensorboard  # only needed once (e.g., at top of notebook)

%tensorboard --logdir logs

Screenshot of TensorBoard running in Colab

Вам также не нужно устанавливать хост и порт (хотя вы можете; он должен работать до тех пор, пока хост разрешается), и вам не нужно работать в фоновом режиме. Экземпляр TensorBoard, показанный в записной книжке, будет продолжать обновлять sh в режиме реального времени по мере того, как вы будете записывать дополнительные учебные данные из своей записной книжки Colab.

Для получения дополнительной информации вы можете ознакомиться с документами для «Использование TensorBoard в ноутбуках ».

(Раскрытие: я работаю над TensorBoard и реализовал эту функцию.)

...