title: ResourceExhaustedError: OOM when allocating tensor with shape[973771,128] and type float on /job:localhost/replica:0/task:0/
body: <blockquote>
    <p><a href="https://tensorflow.google.cn/hub/tf2_saved_model" rel="nofollow noreferrer">https://tensorflow.google.cn/hub/tf2_saved_model</a></p>
  </blockquote>
  
  <pre><code>import tensorflow as tf
  import tensorflow_hub as hub
  hub_url = "https://hub.tensorflow.google.cn/google/tf2-preview/nnlm-en-dim128/1"
  embed = hub.KerasLayer(hub_url)
  embeddings = embed(["A long sentence.", "single-word", "http://example.com"])
  print(embeddings.shape, embeddings.dtype)
  </code></pre>
  
  <p>ResourceExhaustedError: OOM when allocating tensor with shape[973771,128] and type float on /job:localhost/replica:0/task:0/device:CPU:0 by allocator cpu [Op:RestoreV2]</p>
  
  <p>Please tell me how to solve, Thanks</p>