是否在数据集.repeat(),数据集.batch()和数据集.prefetch()之前或之后缓存数据集?

时间:2019-02-04 18:32:56

标签: python tensorflow pipeline tensorflow-datasets

我正在使用Dataset API构建数据管道,以便在多个GPU上进行训练。数据集由大量带有多热标签的png图像组成。我是否在批处理之前或在重复,预取之前/之后进行缓存?目的是最大程度地减少IO和内存使用量,同时提高训练速度。

ValueError('malformed node or string:..

0 个答案:

没有答案