0cecfbb6929c9f9b447d5d75a76b362f20a371cf,tensorpack/dataflow/serialize.py,TFRecordSerializer,save,#Any#Any#,147

Before Change

            df (DataFlow): the DataFlow to serialize.
            path (str): output tfrecord file.
        
        if os.environ.get("TENSORPACK_COMPATIBLE_SERIALIZE", "msgpack") == "msgpack":
            def _dumps(dp):
                return dumps(dp)
        else:

After Change

        size = _reset_df_and_get_size(df)
        with tf.python_io.TFRecordWriter(path) as writer, get_tqdm(total=size) as pbar:
            for dp in df:
                writer.write(dumps(dp))
                pbar.update()

    @staticmethod

Instances