"torchvision/vscode:/vscode.git/clone" did not exist on "af225a8ad529a2a337bca2faef6752199f637d38"
Commit 28b47e60 authored by Bruce MacDonald's avatar Bruce MacDonald
Browse files

accurate loading bar for resume download

parent 61e39bf5
......@@ -90,7 +90,7 @@ def download_file(download_url, models_home, location, file_size):
response = requests.get(download_url, headers=header, stream=True)
response.raise_for_status()
total_size = int(response.headers.get('content-length', 0))
total_size = int(response.headers.get('content-length', 0)) + first_byte
with open(local_filename, 'ab' if first_byte else 'wb') as file, tqdm(
total=total_size,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment