Skip to content

v0.0.1
Compare
Choose a tag to compare
@DeepakTatachar DeepakTatachar released this 24 Jun 21:45
· 36 commits to master since this release
v0.0.1
df34a86

First Release of ADV-TRAIN

This is a framework built on top of pytorch to make machine learning training and inference tasks easier. Along with that it also enables easy dataset and network instantiations, visualize boundaries and more.

Read the latest documentation at https://adv-train.readthedocs.io/en/latest/

Why use this framework?

  • It is very easy to use and well documented and tested
  • The framework supports resume (Yes you can restart training from where ever you left off when your server crashed!).
  • The framework also implements support for train/validation splits of your choice with early stopping baked in.
  • Single argument change for using different datasets and models i.e. convinence at you fingertips
  • Dataloader parameters optimized for highest possbile performance when traning.
  • Supports multi-gpu training (single parameter update required)
Assets 2