mirror of
https://github.com/hanzo-ml/katib.git
synced 2026-10-30 14:46:14 +00:00
4ddca17120
* support JSON format logs in file-metrics-collector * review: convert fileFormat to type FileSystemFileFormat * Update cmd/metricscollector/v1beta1/file-metricscollector/main.go Co-authored-by: Andrey Velichkevich <[email protected]> * review: remove func (f FileSystemFileFormat) String() * review: get metricRegList only when the format is TEXT * review: change var name in a script for e2e * review: explict specify the cloudml-hypyertune in the Dockerfile * review: use reflect.DeepEqual instead of go-cmp.Diff * review: stop using 'JSON' directly in error statements * review: install specific version cloudml-hypertune * review: get objType in the updateStopRules function * review: save optimalObjValue across multiple stopRules * review: add warning messages to parseTimestamp func * review: generate test files with go test command * review: change api for new feature Co-authored-by: Andrey Velichkevich <[email protected]>
770 B
770 B
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.