ea2a0bfb9f60f7e03a3ed25244622e0d90b275a5,opennmt/inputters/record_inputter.py,,write_sequence_record,#Any#Any#,67

Before Change

    writer: A ``tf.python_io.TFRecordWriter``.
  
  shape = list(vector.shape)
  values = vector.flatten().tolist()

  example = tf.train.Example(features=tf.train.Features(feature={
      "shape": tf.train.Feature(int64_list=tf.train.Int64List(value=shape)),

After Change

  
  feature_list = tf.train.FeatureList(feature=[
      tf.train.Feature(float_list=tf.train.FloatList(value=values))
      for values in vector.astype(np.float32)])
  feature_lists = tf.train.FeatureLists(feature_list={"values": feature_list})
  example = tf.train.SequenceExample(feature_lists=feature_lists)
  writer.write(example.SerializeToString())

Instances