waitwaitforget / swingpytorch

Get into Swing of Pytorch. A flexible and efficient framework for experiments and idea-exploring.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

swingpytorch

Get into Swing of Pytorch. A flexible and efficient framework for experiments and idea-exploring. The objective of this repository is to create a more user-friendly framework for researchers or student who just begin to learn deep learning. There are many existing similar respositories on github, what's the features of swingpytorch?

  • Efficient
  • General and easy-customized

Getting Started to train any deep learning models

Data preparation

Take image classification for example, suppose you are going to train a classifier on you own image dataset. There're several steps you need to follow.

  • download (place) your data to a specific folder
  • split train/val/test dataset

About

Get into Swing of Pytorch. A flexible and efficient framework for experiments and idea-exploring.


Languages

Language:Python 100.0%