mirror of
https://github.com/hanzo-ml/katib.git
synced 2026-09-24 00:31:27 +00:00
Ignas BaranauskasandGitHub
af0f32db70
* Chnage the isort profile to black, and add pkg dir for black and flake8 Signed-off-by: Ignas Baranauskas <[email protected]> * Fix the formating Signed-off-by: Ignas Baranauskas <[email protected]> * Fix flake8 lint issues Signed-off-by: Ignas Baranauskas <[email protected]> --------- Signed-off-by: Ignas Baranauskas <[email protected]>
PyTorch MNIST Image Classification Example
This is PyTorch MNIST image classification training container with saving metrics to the file or printing to the StdOut. It uses convolutional neural network to train the model.
Katib uses this training container in some Experiments, for instance in the file Metrics Collector example, the file Metrics Collector with logs in JSON format example, the median stopping early stopping rule with logs in JSON format example and the PyTorchJob example.