Skip to content

belskikh/kekas

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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