361d87cc6d4fe86c82204becce00f4d595e1c459,open_seq2seq/utils/helpers.py,,_restore_embed,#Any#Any#Any#,415

Before Change

def _restore_embed(embed_var, var_to_shape_map, reader):
  has_embed = len([var for var in var_to_shape_map if "EmbeddingMatrix" in var]) > 0
  if has_embed:
    return None, False // assume same name
  for var in var_to_shape_map:
    if var.endswith("dense/kernel") and var_to_shape_map[var] == tf.transpose(embed_var).shape:
      print("Assigning", var, "to", embed_var.name)
      return embed_var.assign(reader.get_tensor(var).T), True

After Change

      print("Assigning", var, "to", embed_var.name)
      tensor = reader.get_tensor(var).T
      if tensor.dtype != var.dtype.as_numpy_dtype():
        return embed_var.assign(tf.cast(tensor, embed_var.dtype)), True
      return embed_var, False
  return None, None

Instances