move to azure

This commit is contained in:
Jeff Wu 2021-02-11 13:10:14 -08:00
parent ddfecb3932
commit d6f4e2956b

View File

@ -17,7 +17,7 @@ for ds in [
]: ]:
for split in ['train', 'valid', 'test']: for split in ['train', 'valid', 'test']:
filename = ds + "." + split + '.jsonl' filename = ds + "." + split + '.jsonl'
r = requests.get("https://storage.googleapis.com/gpt-2/output-dataset/v1/" + filename, stream=True) r = requests.get("https://openaipublic.blob.core.windows.net/gpt-2/output-dataset/v1/" + filename, stream=True)
with open(os.path.join(subdir, filename), 'wb') as f: with open(os.path.join(subdir, filename), 'wb') as f:
file_size = int(r.headers["content-length"]) file_size = int(r.headers["content-length"])