Repository files navigation

• Pytorch Image Classification with Tensorboard
  • Quick Links
  • About
  • Setup
    • Installation
    • Training

This repo contains tutorials covering how to do image classification usingPyTorch,TorchVision andTensorBoard using Python 3.7.

If you find any mistakes or disagree with any of the explanations, please do not hesitate tosubmit an issue. I welcome any feedback, positive or negative!

1. Download the GitHub repo by using the following command running from the terminal.

git clone https://github.com/arpanmukherjee/Pytorch-image-classifier-with-Tensorboardcd Autoencoders-and-more-using-PyTorch/

2. Installpip from the terminal, for more details please lookhere. Go to the following project folder and install all the dependencies by running the following command. By running this command, it will install all the dependencies you will require to run the project.

pip install -r requirements.txt

The network can be trained usingmain.py script. Currently, it only accepts the following arguments with the allowed values. Please strictly follow the argument name and any of the values.

Arguments that have no default value, you must provide value to run the script.

python main.py --dataset STL10 --use-cuda True

If you think the model is taking too much time, you can consider using GPU. Setuse_cuda argument asTrue.