gpt4 book ai didi

tensorflow - tf.Dataset.from_tensor_slices性能问题

转载 作者:行者123 更新时间:2023-11-30 09:17:16 25 4
gpt4 key购买 nike

对于我的输入 fn,我正在填充 tf.Data.Dataset.from_tensor_slices((np.array1, np.array2, np.array3)

我通过调用 dataset.map 来解析它们。结果我返回了我的数据集。

我知道数据集的初始化可能很慢,但是当我调用 tf.estimator train_and_evaluate 函数来训练和评估模型时,它的性能非常糟糕。可能它正在填充每个时期的数据。原因是什么?我怎样才能克服这个问题。您有什么建议?

谢谢。

最佳答案

问题是您正在执行 C++ Python 往返。如果你能尽量避免使用 numpy 。也许尝试使用 tf 记录。

您在 map 功能中到底在做什么?

关于tensorflow - tf.Dataset.from_tensor_slices性能问题,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52345867/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com