pytorch-mnist

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.