wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-03 13:10:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

[SGD] Imagenet example (basic) (#8020)

Browse Source 
* Checkpoint the image-models example

* Update cluster definition

* Fix copyright info

* Use original args

* Checkpoint fixes

* Add README

* Add some missing features

* Format

* Get rid of the unused Namespace class

* Address comments

* Link the imagenet example in docs

* Cleanup

* Fix lint
This commit is contained in:
Maksim Smolin
2020-04-17 13:33:55 -07:00
committed by GitHub
parent 90ef585fd5
commit d6f4e5b3e1
9 changed files with 989 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/source/raysgd/raysgd_pytorch.rst
+3
View File
@@ -717,6 +717,9 @@ to contribute an example, feel free to create a `pull request here <https://gith
- `Semantic Segmentation example <https://github.com/ray-project/ray/blob/master/python/ray/util/sgd/torch/examples/segmentation/train_segmentation.py>`__:
Fine-tuning a ResNet50 model on VOC with Batch Norm.
- `ImageNet Models example <https://github.com/ray-project/ray/blob/master/python/ray/util/sgd/torch/examples/image_models/train.py>`__:
Training state-of-the-art ImageNet models.
- `CIFAR10 example <https://github.com/ray-project/ray/blob/master/python/ray/util/sgd/torch/examples/cifar_pytorch_example.py>`__:
Training a ResNet18 model on CIFAR10.