torch-toolbox

PyTorch utilities

A collection of reusable utility functions and classes to simplify PyTorch development

🛠 Toolbox to extend PyTorch functionalities

GitHub

417 stars
12 watching
56 forks
Language: Python
last commit: 6 months ago
arclossautoaugmentcosinewarmuplrcoslosscutoutcv2flopsinitializersl2softmaxlabelsmoothinglmdblmdb-datasetlookaheadmetricsmixuppytorchswishswitchnormteansformtoolbox

Related projects:

Repository Description Stars
vermeille/torchelie A collection of utility functions and tools for building deep learning models with PyTorch 111
mrdrozdov-github/pytorch-extras A collection of utility functions to extend PyTorch functionality 50
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,663
nearai/torchfold A PyTorch module for dynamic batching and optimized computation on deep neural networks 221
pytorch-labs/torchfix A tool to analyze and fix issues in PyTorch-related Python code, with automated fixes available. 102
bloodaxe/pytorch-toolbelt A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. 1,520
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
e-lab/torch-toolbox A collection of reusable components and utilities for building Torch-based projects 199
pytorchbearer/torchbearer A PyTorch model fitting library designed to simplify the process of training deep learning models. 636
clcarwin/convert_torch_to_pytorch Converts PyTorch models to their Torch equivalents. 539
t-vi/pytorch-tvmisc A collection of utilities and tools for building and improving deep learning models in PyTorch 468
amdegroot/pytorch-containers A collection of PyTorch implementations of Torch Table Layers to simplify the transition from old Torch architectures. 89
metaopt/torchopt An efficient library for differentiable optimization built on top of PyTorch. 544
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
ptrblck/pytorch_misc A collection of code snippets and examples exploring various PyTorch concepts and techniques 544