根據(jù)TensorFlow 文檔,類的prefetch和map方法tf.contrib.data.Dataset都有一個名為的參數(shù)buffer_size。對于prefetchmethod,該參數(shù)稱為,buffer_size并且根據(jù)文檔:buffer_size:一個tf.int64標量tf.Tensor,表示預取時將要緩沖的最大元素數(shù)。對于該map方法,output_buffer_size根據(jù)文檔,該參數(shù)稱為和:output_buffer_size:(可選。)tf.int64標量tf.Tensor,表示將要緩沖的最大已處理元素數(shù)。同樣,對于shuffle方法,根據(jù)文檔顯示相同的數(shù)量:buffer_size:一個tf.int64標量tf.Tensor,表示此數(shù)據(jù)集中要從中采樣新數(shù)據(jù)集的元素數(shù)。這些參數(shù)之間有什么關系?假設我創(chuàng)建一個Dataset對象,如下所示: tr_data = TFRecordDataset(trainfilenames) tr_data = tr_data.map(providefortraining, output_buffer_size=10 * trainbatchsize, num_parallel_calls\=5) tr_data = tr_data.shuffle(buffer_size= 100 * trainbatchsize) tr_data = tr_data.prefetch(buffer_size = 10 * trainbatchsize) tr_data = tr_data.batch(trainbatchsize)buffer以上代碼段中的參數(shù)在扮演什么角色?
- 3 回答
- 0 關注
- 7020 瀏覽
添加回答
舉報
0/150
提交
取消