Skip to content

Commit

Permalink
Gluon ReRun
Browse files Browse the repository at this point in the history
Reran gluon on Azure DSVM and updated results. Fixed common\utils.py
since "!ls" would break linux install
  • Loading branch information
Ilia Karmanov committed Jun 19, 2018
1 parent 29cf517 commit 671e2a0
Show file tree
Hide file tree
Showing 4 changed files with 126 additions and 100 deletions.
5 changes: 2 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ Input for this model is the standard [CIFAR-10 dataset](http://www.cs.toronto.ed
| [Keras(TF)](notebooks/Keras_TF_MultiGPU.ipynb) | 38min | 18min |
| [Tensorflow](notebooks/Tensorflow_MultiGPU.ipynb) | 33min | 22min |
| [Chainer](notebooks/Chainer_MultiGPU.ipynb) | 29min | 8min |
| [MXNet(Gluon)](notebooks/Gluon_MultiGPU.ipynb) | 27min | 8min |
| [MXNet(Gluon)](notebooks/Gluon_MultiGPU.ipynb) | 29min | 10min |

**Train w/ synthetic-data in RAM**

Expand All @@ -65,12 +65,11 @@ Input for this model is the standard [CIFAR-10 dataset](http://www.cs.toronto.ed
| [Keras(TF)](notebooks/Keras_TF_MultiGPU.ipynb) | 36min | 15min |
| [Tensorflow](notebooks/Tensorflow_MultiGPU.ipynb) | 25min | 14min |
| [Chainer](notebooks/Chainer_MultiGPU.ipynb) | 27min | 7min |
| [MXNet(Gluon)](notebooks/Gluon_MultiGPU.ipynb) | 26min | 7min |
| [MXNet(Gluon)](notebooks/Gluon_MultiGPU.ipynb) | 28min | 8min |

Notes:

1. Chainer suffered an AUC drop relative to all other frameworks when going from single to multi-GPU
2. Gluon example at time of writing does not seem to work on Azure VMs and the notebook is provided kindly by Thomas who has run it on AWS

Input for this model is 112,120 PNGs of chest X-rays resized to (264, 264). **Note for the notebook to automatically download the data you must install [Azcopy](https://docs.microsoft.com/en-us/azure/storage/common/storage-use-azcopy-linux#download-and-install-azcopy) and increase the size of your OS-Disk in Azure Portal so that you have at-least 45GB of free-space (the Chest X-ray data is large!). The notebooks may take more than 10 minutes to first download the data.** These notebooks train DenseNet-121 and use native data-loaders to pre-process the data perform some augmentations (random horizontal flip and random crop to 224px).

Expand Down
Loading

0 comments on commit 671e2a0

Please sign in to comment.