-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Fix broken progress bar #524
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
torchvision/datasets/utils.py
Outdated
@@ -45,15 +47,22 @@ def download_url(url, root, filename, md5): | |||
if os.path.isfile(fpath) and check_integrity(fpath, md5): | |||
print('Using downloaded and verified file: ' + fpath) | |||
else: | |||
pbar_ctor = tqdm_notebook if '__IPYTHON__' in locals() else tqdm |
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
This comment was marked as off-topic.
This comment was marked as off-topic.
Sorry, something went wrong.
- Fix broken update calculation - Make progress bar use the neat `unit_scale` feature of tqdm
Thanks @vfdev-5, good to know. But as I mentioned, the The plain progress bar also works decently in notebooks I think. |
See tqdm/tqdm#394 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, thanks!
- Fix broken update calculation - Make progress bar use the neat `unit_scale` feature of tqdm
Some minor fixes to make the downloads look nicer:
unit_scale
featureof tqdm
Usenot working, removedtqdm_notebook
in an IPython environment