基於PyTorch的實用Library

本文紀錄一些基於PyTorch的深度學習Library

pprp/SimpleCVReproduction

pprp/awesome-attention-mechanism-in-cv

facebookresearch/detectron2

jacobgil/pytorch-grad-cam

xmu-xiaoma666/External-Attention-pytorch

from attention.SEAttention import SEAttention
import torch

input=torch.randn(50,512,7,7)
se = SEAttention(channel=512,reduction=8)
output=se(input)
print(output.shape)

thuml/Transfer-Learning-Library

facebookresearch/DomainBed

python3 -m domainbed.scripts.train\
--data_dir=./domainbed/data/MNIST/\
--algorithm IGA\
--dataset ColoredMNIST\
--test_env 2

TorchSSL/TorchSSL

adobe/antialiased-cnns

alexandrosstergiou/adaPool

rentainhe/pytorch-pooling

KevinMusgrave/pytorch-metric-learning

MadryLab/robustness

BorealisAI/advertorch

CoinCheung/pytorch-loss

jettify/pytorch-optimizer

bonlime/pytorch-tools

zhangyongshun/BagofTricks-LT

libffcv/ffcv

ShangHua-Gao/RBN

chou141253/FGVC-PIM

navervision/proxy-synthesis

Guide to Pytorch Learning Rate Scheduling

DefTruth/torchlm

lightly-ai/lightly

--

--

Machine Learning | Deep Learning | https://linktr.ee/yanwei

Love podcasts or audiobooks? Learn on the go with our new app.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store