Как вы можете использовать Google Colab TPU для сохранения контрольных точек в облачном хранилище Google в Tensorflow 2.0? - PullRequest
Новая
       39

Как вы можете использовать Google Colab TPU для сохранения контрольных точек в облачном хранилище Google в Tensorflow 2.0?

0 голосов
/ 17 января 2020
  1. Сценарии в Tensorflow 1.X, но они все еще не работают в Tensorflow 2.0

2.Как я могу перенести эти сценарии в Tensorflow 2.0 


import datetime
import json
import os
import pprint
import random
import string
import sys
import tensorflow as tf
import tensorflow_io as tfio

assert 'COLAB_TPU_ADDR' in os.environ, 'ERROR: Not connected to a TPU runtime; please see the first cell in this notebook for instructions!'
TPU_ADDRESS = 'grpc://' + os.environ['COLAB_TPU_ADDR']
print('TPU address is', TPU_ADDRESS)

from google.colab import auth
auth.authenticate_user()

with tf.compat.v1.Session(TPU_ADDRESS) as session:
  print('TPU devices:')
  pprint.pprint(session.list_devices())

  # Upload credentials to TPU.
  with open('/content/adc.json', 'r') as f:
    auth_info = json.load(f)
  tf.contrib.cloud.configure_gcs(session, credentials=auth_info)
#   Now credentials are set for all future sessions on this TPU.
...