examples/pytorch_mnist/ks_app
David Sabater Dinter a630fcea34 [mnist_pytorch] fix train image (#342)
* Default to model trained with CPUs
TODO: Enable A/B testing with Seldon to load GPU and CPU models

* Checkout 1.0rc1 release as latest Pytorch master seems to have MPI backend detection broken

* Track changes in pytorch_mnist/training/ddp/mnist folder to trigger test jobs

* Repoint to pull images from gcr.io/kubeflow-ci built during pre-submit

* Fix image webui name

* Fix logging

* Add GCFS to CPU train

* Fix logging

* Add GCFS to CPU train

* Default to model trained with GPUs
TODO: Enable A/B testing with Seldon to load GPU and CPU models

* Fix Predict() method as Seldon expects 3 arguments

* Fix x reference
2018-11-24 13:22:28 -08:00
..
components [mnist_pytorch] fix train image (#342) 2018-11-24 13:22:28 -08:00
environments E2E Pytorch mnist example (#274) 2018-11-18 14:24:43 -08:00
.gitignore E2E Pytorch mnist example (#274) 2018-11-18 14:24:43 -08:00
app.yaml E2E Pytorch mnist example (#274) 2018-11-18 14:24:43 -08:00