Skip to content

[BETA] Easy-to-use pytorch model trainer, dataloader creators, and utilities + my own implementations of select models

Notifications You must be signed in to change notification settings

dansuh17/pytorch-land

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch Land

This is a repository of select implementations of deep learning models using pytorch.

There is also my own mini-framework for model training (which I call simply NetworkTrainer), that ended up with something very similar to ignite, created in order to reduce common boilerplating. More flexible and intuitive than ignite, in my opinion :).

More models to be added, and improvements on NetworkTrainer is under way.

Installation

pip3 install pytorch-land

See pypi page for package details.

Implemented Models

CNN architectures

GANs

Autoencoders

  • Stacked Denoising Autoencoders [paper] [code]
  • Stacked Convolutional Denoising Autoencoders (2017) [paper] [code]

Requirements

Required packages are specified in requirements.txt file. The packages can be installed using the following command:

pip3 install -r requirements.txt

Notably, the codes are compatible with pytorch 0.4 - working on with pytorch 1.1 compatibility.

NetworkTrainer

Datasets

About

[BETA] Easy-to-use pytorch model trainer, dataloader creators, and utilities + my own implementations of select models

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages