为了使用Docker容器在AI平台上训练tensorflow keras模型,我们使用tf.data.Dataset将存储在GCS上的原始图像转换为tfrecord。因此,数据永远不会在本地存储。相反,原始图像直接转换为tfrecords到另一个桶。是否可以利用凯德罗和tfrecord数据集以及tf.data.Dataset的流功能?根据文档,kedro似乎不支持tfrecord数据集。
tf.data.Dataset
发布于 2020-07-31 08:41:56
目前只有TF相关的数据集是TensorFlowModelDataset (dataset.html),但是您可以轻松地使用添加您自己的自定义数据集,或者请添加一个功能请求/您在回购中的贡献。
TensorFlowModelDataset
https://stackoverflow.com/questions/63182406
相似问题