日々精進

新しく学んだことを書き留めていきます

TensorFlowにS3のパスを渡すと読み込んでくれる

dataset = tf.data.TFRecordDataset(filenames=s3_paths, num_parallel_reads=num_parallel_calls)

のように、S3オブジェクトのパス(s3://~)を渡すと、TF側でS3からデータをダウンロードしてくれる。便利。

参考:

github.com

tensorflow.classcat.com