nitrain / nitrainLinks
Train AI models efficiently on medical images using any framework
☆1,870Updated last year
Alternatives and similar repositories for nitrain
Users that are interested in nitrain are comparing it to the libraries listed below
Sorting:
- A lightweight library for PyTorch training tools and utilities☆1,700Updated this week
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,835Updated last year
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,049Updated last year
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,950Updated last week
- Neural network visualization toolkit for keras☆2,991Updated 3 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,671Updated last week
- 3.8% and 18.3% on CIFAR-10 and CIFAR-100☆1,308Updated 5 years ago
- Research Framework for easy and efficient training of GANs based on Pytorch☆1,421Updated 2 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Updated 3 years ago
- Collection of generative models in Pytorch version.☆2,633Updated 5 years ago
- Improving Convolutional Networks via Attention Transfer (ICLR 2017)☆1,455Updated 6 years ago
- ☆1,209Updated 5 years ago
- Stochastic Weight Averaging in PyTorch☆973Updated 3 years ago
- Python library to easily log experiments and parallelize hyperparameter search for neural networks☆736Updated 2 years ago
- A PyTorch implementation of the NIPS 2017 paper "Dynamic Routing Between Capsules".☆1,742Updated 6 years ago
- Log TensorBoard events without touching TensorFlow☆627Updated 2 years ago
- PyTorch Boilerplate For Research☆608Updated 3 years ago
- Some custom dataset examples for PyTorch☆868Updated 5 years ago
- A CV toolkit for my papers.☆2,049Updated 6 months ago
- A state-of-the-art semi-supervised method for image recognition☆1,625Updated 4 years ago
- PyTorch and Tensorflow functional model definitions☆587Updated 7 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,917Updated last year
- ☆3,228Updated 6 years ago
- A small package to create visualizations of PyTorch execution graphs☆3,384Updated 6 months ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,304Updated 2 months ago
- Code for the Lovász-Softmax loss (CVPR 2018)☆1,391Updated last year
- torchbearer: A model fitting library for PyTorch☆640Updated last year
- A memory-efficient implementation of DenseNets☆1,537Updated 2 years ago
- Semantic Segmentation Architectures Implemented in PyTorch☆3,416Updated last year
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,093Updated 3 years ago