Upcoming Features

  •  Plot Feature For Multi Model Training
  • Support for Imbalanced Datasets (simple and fast handling for unbalanced datasets)
  • Model Compression
  • Custom TFRecords Formats
  •  Custom Model Creation 
  • Improvisation in training loop​​
  • Test Time Augmentations
  • SOTA Augmentations
  • Label Smoothing
  • Smoother training with option of batch wise or epoch wise LR update.
  • Addition of Logging facilities
  • GRADCAM for model output visualization & debugging
  • Support for ResNext, SeResNet, SeResNext, NasNet (all variants).
  • Wide range of evaluation metrics.