tensorflow-dataset 内网下载 指定目录

内网下载报错

解决办法是设置环境变量，指向你的代理服务器TFDS_HTTP_PROXY=http://xxx、TFDS_HTTPS_PROXY=http://xxx。

留意到，赋值的是你的代理服务器，且最好协议都使用http(即使TFDS_HTTPS_PROXY也要使用http协议连服务器)。如果不这么做，可能会遇到ProxyError

指定目录

如果你的数据集创建是通过tfds.builder创建的，可以使用以下参数，指定数据集下载到指定目录：

dataset_builder = tfds.builder(config.dataset, data_dir='/home/xxx/dataset/tensorflow-datasets')
dataset_builder.download_and_prepare(download_dir='/home/xxx/dataset/tensorflow-datasets/downloads')