Skip to content
This repository was archived by the owner on Jul 1, 2025. It is now read-only.

Conversation

goldsborough
Copy link
Contributor

Improves the script to download datasets a bit.

  1. Adds an argument parser:
 psag@psag-mbp:~/home/Glow python utils/download_test_db.py -h                                                                                                                                                                                                                                                           
usage: download_test_db.py [-h]
                           [-d {ptb,mnist,cifar10} [{ptb,mnist,cifar10} ...]]
                           [-a]

Download datasets for Glow

optional arguments:
  -h, --help            show this help message and exit
  -d {ptb,mnist,cifar10} [{ptb,mnist,cifar10} ...], --datasets {ptb,mnist,cifar10} [{ptb,mnist,cifar10} ...]
  -a, --all

This allows downloading individual datasets with -d <dataset1> [, <dataset2>] or all of them with -a/--all. Closes #211.

Also adds a progress bar and better error handling and descriptive output:

screen shot 2018-01-02 at 22 41 35

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants