Toy example of a python script that instantiates and trains a PyTorch neural network on the FashionMNIST dataset with several common and useful featur
This simple_pytorch_example project is a toy example of a python script that instantiates and trains a PyTorch neural network on the FashionMNIST dataset with several common and useful features: Choose between two different neural network architectures Make architectures parametrizable Read input arguments from config file or command line (command line arguments override config file ones) Download FashionMNIST dataset if not already downloaded Monitor training progress on the terminal and/or with TensorBoard logs Accuracy, loss, confusion matrix More details about FashionMNIST […]
Read more