torchbearer

Model trainer

A PyTorch model fitting library designed to simplify the process of training deep learning models.

torchbearer: A model fitting library for PyTorch

GitHub

636 stars
25 watching
65 forks
Language: Python
last commit: about 1 year ago
deep-learningdifferentiable-programmingmachine-learningmodel-fittingpython3pytorch

Related projects:

Repository Description Stars
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,822
ahmedfgad/torchga Trains PyTorch models using a genetic algorithm 96
ne7ermore/torch-light A comprehensive collection of deep learning examples using PyTorch, covering various applications and models. 537
vermeille/torchelie A collection of utility functions and tools for building deep learning models with PyTorch 111
clcarwin/convert_torch_to_pytorch Converts PyTorch models to their Torch equivalents. 541
hughperkins/pytorch Provides Python wrappers for PyTorch and Lua, enabling developers to use PyTorch's deep learning capabilities from both languages. 432
torch/cutorch Provides a CUDA backend for the PyTorch deep learning framework 337
pytorch/tnt A collection of tools and utilities for building and training neural networks with PyTorch. 1,669
nearai/torchfold A PyTorch module for dynamic batching and optimized computation on deep neural networks 221
bloodaxe/pytorch-toolbelt A comprehensive Python library with PyTorch extensions for rapid prototyping and machine learning model development. 1,523
huggingface/pytorch-openai-transformer-lm Implementing OpenAI's transformer language model in PyTorch with pre-trained weights and fine-tuning capabilities 1,511
fanq15/caffe_to_torch_to_pytorch A tool for converting models between Caffe and PyTorch 150
rdspring1/pytorch_gbw_lm Trains a large-scale PyTorch language model on the 1-Billion Word dataset 123
ahirner/pytorch-retraining An experiment and benchmarking framework for evaluating the effectiveness of transfer learning in PyTorch-based deep learning models 170
dmarnerides/pydlt A PyTorch-based toolbox for building and training deep learning models with ease. 204