Skip to content

belskikh/kekas

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Kekas

Kek it easy.

Kekas is a simple tool for training neural networks on Pytorch.

I tried to keep it as simple as possible.

Installation

pip install kekas

Mixed Precision requirements

Kekas use https://github.com/NVIDIA/apex library for mixed precision training, so follow the installation instructions from its repo.

Features

  • Awesome name
  • Mixed Precision (FP16)
  • Learning Rate Finder
  • One Cycle policy
  • Tensoboard logging
  • Best checkpoints saving
  • Early stopping
  • TTA
  • Freeze / unfreeze
  • Easy customization

Quick start guide

I don't beieve in quick start guides, I think that they create more questions than answers.

Instead, I've created a detailed Tutorial notebook. Read it.Open In Colab

Contribution guide

Just contribute something good and don't contribute anything bad.

Citing

If you find this library useful for your research, please consider citing:

@misc{aleksandr belskikh_2019,
    author      = {Aleksandr Belskikh},
    title       = {{kekas: Just another DL library}},
    month       = dec,
    year        = 2019,
    doi         = {10.5281/zenodo.2577861},
    version     = {0.1.23},
    publisher   = {Zenodo},
    url         = {https://doi.org/10.5281/zenodo.2577861},
    }

DOI