Skip to content

Instantly share code, notes, and snippets.

@hanneshapke

hanneshapke/cast_between_tft_tfhub.py

Created March 9, 2020 18:00
Show Gist options
  • Save hanneshapke/0ce64b4f90207d299a57614ffb8fa80e to your computer and use it in GitHub Desktop.
Save hanneshapke/0ce64b4f90207d299a57614ffb8fa80e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
cast_between_tft_tfhub.py
input_word_ids = tf.cast(inputs["input_word_ids"], dtype=tf.int32)
input_mask = tf.cast(inputs["input_mask"], dtype=tf.int32)
input_type_ids = tf.cast(inputs["input_type_ids"], dtype=tf.int32)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment