lilujunai Goto Github PK

pytorch-hardnet

35% faster than ResNet: Harmonic DenseNet, A low memory traffic network

pytorch-hdml

Hardness-Aware Deep Metric Learning (CVPR2019) in pytorch

pytorch-hebbian

A lightweight and flexible framework for Hebbian learning in PyTorch.

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- (SE)ResNet/ResNeXT, DPN, EfficientNet, MixNet, MobileNet-V3/V2/V1, MNASNet, Single-Path NAS, FBNet, and more

pytorch-imagenet-cifar-coco-voc-training

Training examples and results for ImageNet/CIFAR/COCO/VOC training.Image Classification/Object Detection.

pytorch-lamb

Implementation of https://arxiv.org/abs/1904.00962

pytorch-lars

PyTorch implementation of LARS (Layer-wise Adaptive Rate Scaling)

pytorch-loss

label-smooth, amsoftmax, focal-loss, triplet-loss. Maybe useful

pytorch-lr-finder

A learning rate range test implementation in PyTorch

pytorch-memonger

Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174

pytorch-memory-utils

pytorch memory track code

pytorch-meta

A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch

pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

pytorch-moonshine

Cheap distillation for convolutional neural networks.

pytorch-multi-label-classifier

A pytorch implemented classifier for Multiple-Label classification

pytorch-multi-task-multi-class-classification

旨在搭建一个分类问题在Pytorch框架下的通解，批量解决单任务多分类问题、多任务多分类问题。

pytorch-multigpu

Multi GPU Training Code for Deep Learning with PyTorch

pytorch-multilabel-classification-baseline

a strong multi-label classification baseline

pytorch-opcounter

Count the MACs / FLOPs of your PyTorch model.

pytorch-optimizer

torch-optimizer -- collection of optimizers for Pytorch

pytorch-pcgrad

Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"

pytorch-playground

Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)

pytorch-profiling-tool

pytorch-prunes

Code for https://arxiv.org/abs/1810.04622

pytorch-randaugment

Unofficial PyTorch Reimplementation of RandAugment.

pytorch-rnn-util

A simple PyTorch LSTM package including a general LSTM frame

pytorch-scalable-neural-networks

A pytorch implement of scalable neural netowrks.

pytorch-segmentation-toolbox

PyTorch Implementations for DeeplabV3 and PSPNet

pytorch-self-distillation-final

pytorch-single-path-one-shot-nas

Single Path One-Shot NAS MXNet implementation with Supernet training and searching