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. 40
graal-research/poutyne A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities 572
bearpaw/pytorch-classification An open-source project providing a unified interface for training and evaluating classification models using PyTorch. 1,699
norse/norse Deep learning framework for spiking neural networks 685
neuralhydrology/neuralhydrology A Python library for training neural networks with focus on hydrological applications using PyTorch. 372
chingyaoc/pytorch-reinforce A PyTorch implementation of the REINFORCE algorithm for reinforcement learning in continuous and discrete environments. 266
aiqm/torchani A PyTorch implementation of a neural network potential for molecular simulations 471
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,236
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,044
ramon-oliveira/aorun A deep learning framework on top of PyTorch for building neural networks. 61
ahmedfgad/torchga Trains PyTorch models using a genetic algorithm 96
baguasys/bagua A framework for accelerating PyTorch deep learning training 876
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,729