pywick

Training framework

A PyTorch-based neural network training framework with advanced features and utilities

High-level batteries-included neural network training library for Pytorch

GitHub

398 stars
15 watching
39 forks
Language: Python
last commit: almost 3 years ago
classificationdeep-learningneural-networkspytorchsemantic-segmentation

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
graal-research/poutyne A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities 569
bearpaw/pytorch-classification An open-source project providing a unified interface for training and evaluating classification models using PyTorch. 1,692
norse/norse Deep learning framework for spiking neural networks 673
neuralhydrology/neuralhydrology A Python library for training neural networks with focus on hydrological applications using PyTorch. 364
chingyaoc/pytorch-reinforce A PyTorch implementation of the REINFORCE algorithm for reinforcement learning in continuous and discrete environments. 264
aiqm/torchani A PyTorch implementation of a neural network potential for molecular simulations 464
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
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
ramon-oliveira/aorun A deep learning framework on top of PyTorch for building neural networks. 61
ahmedfgad/torchga Trains PyTorch models using the Genetic Algorithm 95
baguasys/bagua A framework for accelerating PyTorch deep learning training 877
project-dc/pygeneses A PyTorch-based framework for training and studying artificial species in bio-inspired environments 72
nearai/torchfold A PyTorch module for dynamic batching and optimized computation on deep neural networks 221
zijundeng/pytorch-semantic-segmentation Provides PyTorch implementations of various models and pipelines for semantic segmentation in deep learning. 1,724