[PyTorch] Getting Started with PyTorch

Introduction:

PyTorch is a very powerful deep learning framework in python which has a strong GPU acceleration support. Previously, Torch 7 a a good framework based on lua, which is a language few of us like to use. PyTorch is based on Python which has all advantages of the Python such as extendability and easy-to-use feature. It has become popular in many deep learning research topics and well used by many famous research group.

Environment:

System: Ubuntu 16.04 LTS

Python: Anaconda Python 3.5

CUDA Version: 8.0

Installation:

HomePage: pytorch.org

Install command:


conda install pytorch torchvision -c pytorch

Good Example Code:

Generative models:

https://github.com/wiseodd/generative-models

This GitHub repository is a collection of many popular generative models written in PyTorch. They simplify the models and remain only the key parts such as model structure, loss function and training procedure.

CycleGAN and pix2pix:

https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

This GitHub repository looks nice and extendable. They build nice base model and dataloader which could be modified to your own model. Moreover, they use visdom created by Facebook (GitHub) to visualize the training results.

 

Advertisements

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out /  Change )

Google photo

You are commenting using your Google account. Log Out /  Change )

Twitter picture

You are commenting using your Twitter account. Log Out /  Change )

Facebook photo

You are commenting using your Facebook account. Log Out /  Change )

Connecting to %s

Blog at WordPress.com.

Up ↑

%d bloggers like this: