Don't clobber existing files

Author: akx

download_dataset.py

@@ -17,9 +17,17 @@
         filename = ds + "." + split + '.jsonl'
         r = requests.get("https://storage.googleapis.com/gpt-2/output-dataset/v1/" + filename, stream=True)
         r.raise_for_status()
+        file_size = int(r.headers["content-length"])
+        filepath = os.path.join(subdir, filename)
+        try:
+            if os.stat(filepath).st_size == file_size:
+                print('%s already exists and is the expected %d bytes, not redownloading' % (filepath, file_size))
+                r.close()
+                continue
+        except OSError:  # likely "file not found" or similar
+            pass
 
-        with open(os.path.join(subdir, filename), 'wb') as f:
-            file_size = int(r.headers["content-length"])
+        with open(filepath, 'wb') as f:
             with tqdm(ncols=100, desc="Fetching " + filename, total=file_size, unit_scale=True) as pbar:
                 for chunk in r.iter_content(chunk_size=4194304):
                     f.write(chunk)