、、、という 3 つの tfrecord ファイルがあるとneg.tfrecordしpos1.tfrecordますpos2.tfrecord。
私が使う
dataset = tf.data.TFRecordDataset(tfrecord_file)
このコードは 3 つの Dataset オブジェクトを作成します。
私のバッチ サイズは 400 で、これには 200 個の neg データ、100 個の pos1 データ、および 100 個の pos2 データが含まれます。目的のデータセットを取得するにはどうすればよいですか?
このデータセット オブジェクトを keras.fit() (Eager Execution) で使用します。
私の tensorflow のバージョンは 1.13.1 です。
以前は、データセットごとに反復子を取得し、データを取得した後に手動で連結しようとしましたが、非効率で GPU 使用率が高くありませんでした。