Skip to content
An introductory tutorial to using PyTorch and setting up your Deep Learning ToolKit.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
content
loaders
models
.gitignore
LICENSE
README.md
main.py
net.py

README.md

Intro To PyTorch

This is the implementation of basic MNIST classification using PyTorch.

Requirements

  • OS/VM of your choice
  • Anaconda

Environment

Setup your Anaconda Environment using the following commands -

$ conda create -n oml python=3.6
$ conda activate oml
(oml) $ conda install pytorch torchvision -c pytorch
(oml) $ conda install tensorflow-gpu
(oml) $ conda install tensorboardX

Usage

After successfully setting up environment, you can run the code by -

tensorboard --logdir=./runs/ --host 0.0.0.0 --port 6007 & python main.py --phase train --continue_train 0

Now click here to visualize the results - http://localhost:6007

Tutorial

A detailed tutorial about the contents covered in this repo can found at - LINK

Results

Here are the final TensorBoard Results - TensorBoardX

Author

Sri Raghu Malireddi / @r4ghu

You can’t perform that action at this time.