Alternatives and similar repositories for nitrain:

Users that are interested in nitrain are comparing it to the libraries listed below

• pytorch / tnt
  A lightweight library for PyTorch training tools and utilities
  ☆1,683Updated this week
• sksq96 / pytorch-summary
  Model summary in PyTorch similar to `model.summary()` in Keras
  ☆4,033Updated last year
• bckenstler / CLR
  ☆1,207Updated 4 years ago
• waleedka / hiddenlayer
  Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
  ☆1,818Updated last year
• LiyuanLucasLiu / RAdam
  On the Variance of the Adaptive Learning Rate and Beyond
  ☆2,546Updated 3 years ago
• carpedm20 / ENAS-pytorch
  PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"
  ☆2,711Updated last year
• znxlwm / pytorch-generative-model-collections
  Collection of generative models in Pytorch version.
  ☆2,622Updated 4 years ago
• raghakot / keras-vis
  Neural network visualization toolkit for keras
  ☆2,988Updated 3 years ago
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,919Updated 2 months ago
• gpleiss / efficient_densenet_pytorch
  A memory-efficient implementation of DenseNets
  ☆1,529Updated last year
• martinarjovsky / WassersteinGAN
  ☆3,221Updated 6 years ago
• DeepVoltaire / AutoAugment
  Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow
  ☆1,480Updated last year
• szagoruyko / attention-transfer
  Improving Convolutional Networks via Attention Transfer (ICLR 2017)
  ☆1,452Updated 6 years ago
• gram-ai / capsule-networks
  A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".
  ☆1,740Updated 6 years ago
• williamFalcon / test-tube
  Python library to easily log experiments and parallelize hyperparameter search for neural networks
  ☆734Updated 2 years ago
• bamos / densenet.pytorch
  A PyTorch implementation of DenseNet.
  ☆836Updated 6 years ago
• openai / improved-gan
  Code for the paper "Improved Techniques for Training GANs"
  ☆2,307Updated 6 years ago
• pytorchbearer / torchbearer
  torchbearer: A model fitting library for PyTorch
  ☆638Updated last year
• timgaripov / swa
  Stochastic Weight Averaging in PyTorch
  ☆969Updated 3 years ago
• szagoruyko / wide-residual-networks
  3.8% and 18.3% on CIFAR-10 and CIFAR-100
  ☆1,299Updated 5 years ago
• google / compare_gan
  Compare GAN code.
  ☆1,821Updated 4 years ago
• zhoubolei / CAM
  Class Activation Mapping
  ☆1,854Updated 2 years ago
• Luolc / AdaBound
  An optimizer that trains as fast as Adam and as good as SGD.
  ☆2,910Updated last year
• torchgan / torchgan
  Research Framework for easy and efficient training of GANs based on Pytorch
  ☆1,410Updated 2 years ago
• ufoym / imbalanced-dataset-sampler
  A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.
  ☆2,291Updated 2 months ago
• szagoruyko / functional-zoo
  PyTorch and Tensorflow functional model definitions
  ☆585Updated 7 years ago
• wiseodd / generative-models
  Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.
  ☆7,391Updated 11 months ago
• fyu / drn
  Dilated Residual Networks
  ☆1,106Updated 5 years ago
• lessw2020 / Ranger-Deep-Learning-Optimizer
  Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase
  ☆1,197Updated last year
• zhanghang1989 / PyTorch-Encoding
  A CV toolkit for my papers.
  ☆2,047Updated 3 months ago