多次使用dataset.map是个好主意吗?

时间:2019-04-08 03:46:28

标签: tensorflow tensorflow-datasets

我的代码非常慢,因此我试图使其更快。我多次使用tf.data.Dataset.map()方法进行数据扩充:

dataset_1 = tf.data.Dataset.from_tensor_slices(example)

if bool_1 :
    dataset_2 = dataset_1.map(fn_1_t)
else :
    dataset_2 = dataset_1.map(fn_1_f)
if bool_2 :
    dataset_3 = dataset_1.map(fn_2_t)
else :
    dataset_3 = dataset_1.map(fn_2_f)

final_dataset = dataset_3.map(fn_3)

您是否认为使用多个映射不是一个好主意,是否会使CPU无法进行并行计算?

谢谢!

0 个答案:

没有答案