Torchelie

PyTorch toolkit

A collection of utility functions and tools for building deep learning models with PyTorch

Torchélie is a set of utility functions, layers, losses, models, trainers and other things for PyTorch.

GitHub

111 stars
9 watching
11 forks
Language: Python
last commit: 14 days ago
ganlossperceptualpytorchtorchutilsvqvae

Related projects:

Repository Description Stars
bloodaxe/pytorch-toolbelt A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. 1,520
nearai/torchfold A PyTorch module for dynamic batching and optimized computation on deep neural networks 221
pistony/torch-toolbox A collection of reusable utility functions and classes to simplify PyTorch development 417
pytorchbearer/torchbearer A PyTorch model fitting library designed to simplify the process of training deep learning models. 636
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
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,663
metaopt/torchopt An efficient library for differentiable optimization built on top of PyTorch. 544
yu-changqian/torchseg A toolkit for building and training semantic segmentation models using PyTorch. 1,408
zhanghang1989/pytorch-encoding A Python framework for building deep learning models with optimized encoding layers and batch normalization. 2,041
lyhue1991/torchkeras A PyTorch-based model training framework designed to simplify and streamline training workflows by providing a unified interface for various loss functions, optimizers, and validation metrics. 1,782
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
dmarnerides/pydlt A PyTorch-based toolbox for building and training deep learning models with ease. 204
pytorch-labs/torchfix A tool to analyze and fix issues in PyTorch-related Python code, with automated fixes available. 102
torch/cutorch Provides a CUDA backend for the PyTorch deep learning framework 336
mrdrozdov-github/pytorch-extras A collection of utility functions to extend PyTorch functionality 50