/nnlib

PyTorch based NN modules frequently used in various projects

Primary LanguagePythonMIT LicenseMIT

Useful and frequently used code for training neural networks

Requirements:

  • Basic data science libraries: numpy, scipy, tqdm, matplotlib, seaborn, pandas, scikit-learn.
  • We use Pytorch 1.4.0, but higher versions may work too.
  • Additionally, only for extracting data from Tensorboard files one might need tensorflow >= 2.0.