poutyne

Neural net trainer

A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities

A simplified framework and utilities for PyTorch

GitHub

569 stars
17 watching
64 forks
Language: Python
last commit: 5 months ago
Linked from 2 awesome lists

data-sciencedeep-learningkerasmachine-learningneural-networkpythonpytorch

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
erotemic/netharn A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. 39
achaiah/pywick A PyTorch-based neural network training framework with advanced features and utilities 398
neuralhydrology/neuralhydrology A Python library for training neural networks with focus on hydrological applications using PyTorch. 364
ramon-oliveira/aorun A deep learning framework on top of PyTorch for building neural networks. 61
norse/norse Deep learning framework for spiking neural networks 673
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,217
atgambardella/pytorch-es An implementation of an optimization algorithm for training neural networks in machine learning environments. 350
kaixhin/grokking-pytorch An introduction to using PyTorch for deep learning tasks 1,194
4uiiurz1/pytorch-res2net Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. 112
jacobgil/pytorch-pruning This project provides a PyTorch implementation of pruning techniques to reduce the computational resources required for neural network inference. 875
project-dc/pygeneses A PyTorch-based framework for training and studying artificial species in bio-inspired environments 72
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,663
furkanu/deeplearning.ai-pytorch Re-implementation of Coursera's Deep Learning specialization assignments in PyTorch 149
ahmedfgad/torchga Trains PyTorch models using the Genetic Algorithm 95
eduardoleao052/js-pytorch A JavaScript library that provides GPU-accelerated deep learning capabilities with automatic differentiation and neural network layers. 1,084