Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
PyTorch
PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing, primarily developed by Facebook’s AI Research lab.
- GitHub: https://github.com/topics/pytorch
- Wikipedia: https://en.wikipedia.org/wiki/PyTorch
- Repo: https://github.com/pytorch/pytorch
- Created by: Facebook's AI Research lab (FAIR)
- Released: September 2016
- Related Topics: python, pytorch-tutorial,
- Last updated: 2025-02-06 00:24:52 UTC
- JSON Representation
https://github.com/amanchadha/iseebetter
iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press
cnn convolutional-neural-network convolutional-neural-networks gan gans generative-adversarial-network psnr pytorch recurrent-neural-networks residual-networks resnet resnets rnn self-supervised-learning srgan ssim super-resolution unsupervised-learning video-processing video-super-resolution
Last synced: 02 Feb 2025
https://github.com/ShirAmir/dino-vit-features
Official implementation for the paper "Deep ViT Features as Dense Visual Descriptors".
co-segmentation computer-vision deep-learning dino part-segmentation pytorch semantic-correspondence vision-transformers
Last synced: 04 Nov 2024
https://github.com/benedekrozemberczki/APPNP
A PyTorch implementation of "Predict then Propagate: Graph Neural Networks meet Personalized PageRank" (ICLR 2019).
appnp attention deep-learning deep-neural-networks deepwalk gcn graph-attention graph-classification graph-convolutional-neural-networks graph-embedding graph-neural-network iclr machine-learning network-embedding node-embedding node2vec pagerank ppnp pytorch research
Last synced: 08 Nov 2024
https://github.com/HikariTJU/LD
Localization Distillation for Object Detection (CVPR 2022, TPAMI 2023)
deep-learning detection knowledge-distillation ld object-detection pytorch
Last synced: 05 Nov 2024
https://github.com/official-stockfish/nnue-pytorch
Stockfish NNUE (Chess evaluation) trainer in Pytorch
chess deep-learning pytorch stockfish
Last synced: 01 Feb 2025
https://github.com/Jungjee/RawNet
Official repository for RawNet, RawNet2, and RawNet3
extracted-speaker-embeddings pytorch rawnet speaker-embeddings speaker-verification spk-embd voxceleb2
Last synced: 06 Nov 2024
https://github.com/alibaba/lightweight-neural-architecture-search
This is a collection of our zero-cost NAS and efficient vision applications.
automl classfication cvpr2023 deepmad iclr2023 icml-2022 maedet nas neurips-2022 object-detection pytorch training-free-nas zennas zero-cost-nas
Last synced: 15 Nov 2024
https://github.com/wenqifan03/GraphRec-WWW19
Graph Neural Networks for Social Recommendation, WWW'19
deep-social-recommendations graphrec pytorch recommender-systems socal-recommendation social-network
Last synced: 28 Nov 2024
https://github.com/ibm/zshot
Zero and Few shot named entity & relationships recognition
ai deep-learning few-shot few-shot-learning machine-learning named-entity-recognition natural-language-processing natural-language-understanding ned ner nlp nlp-library pytorch relation-extraction relationship-extraction spacy transformer zero-shot zero-shot-learning
Last synced: 01 Feb 2025
https://github.com/KevinMusgrave/pytorch-adapt
Domain adaptation made easy. Fully featured, modular, and customizable.
computer-vision deep-learning domain-adaptation machine-learning pytorch transfer-learning
Last synced: 05 Nov 2024
https://github.com/kakaoenterprise/JORLDY
Repository for Open Source Reinforcement Learning Framework JORLDY
framework gym mlagents pytorch reinforcement-learning
Last synced: 28 Nov 2024
https://github.com/QUVA-Lab/escnn
Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries
Last synced: 13 Nov 2024
https://github.com/uber-research/deeppruner
DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)
iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision
Last synced: 30 Jan 2025
https://github.com/ymfa/seq2seq-summarizer
Pointer-generator reinforced seq2seq summarization in PyTorch
abstractive-summarization attention attention-mechanism coverage coverage-mechanism pointer-network pytorch reinforcement-learning seq2seq summarization summarizer
Last synced: 13 Nov 2024
https://github.com/kefirski/pytorch_RVAE
Recurrent Variational Autoencoder that generates sequential data implemented with pytorch
deep-learning nlp python pytorch vae
Last synced: 02 Nov 2024
https://github.com/quantylab/rltrader
파이썬과 케라스를 이용한 딥러닝/강화학습 주식투자 - 퀀트 투자, 알고리즘 트레이딩을 위한 최첨단 해법 입문 (개정판)
deep-learning keras python pytorch reinforcement-learning tensorflow
Last synced: 31 Jan 2025
https://github.com/quva-lab/escnn
Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
deep-learning equivariance geometric-deep-learning geometry invariant isometries pytorch rotations symmetries
Last synced: 31 Jan 2025
https://github.com/ymcui/PERT
PERT: Pre-training BERT with Permuted Language Model
bert nlp plm pre-trained-model pytorch tensorflow transformers
Last synced: 16 Nov 2024
https://github.com/seeed-studio/modelassistant
Seeed SenseCraft Model Assistant is an open-source project focused on embedded AI. 🔥🔥🔥
arduino deep-learning esp32 image-classification jetson ncnn object-detection onnx openmmlab pytorch raspberry-pi tflite tinyml yolov5
Last synced: 05 Feb 2025
https://github.com/ymcui/pert
PERT: Pre-training BERT with Permuted Language Model
bert nlp plm pre-trained-model pytorch tensorflow transformers
Last synced: 06 Feb 2025
https://github.com/fendouai/pytorch1.0-cn
PyTorch 1.0 官方文档 中文版,欢迎关注微信公众号:磐创AI
pytorch pytorch-nlp pytorch-tutorial
Last synced: 30 Jan 2025
https://github.com/utkuozbulak/pytorch-cnn-adversarial-attacks
Pytorch implementation of convolutional neural network adversarial attack techniques
adversarial-example fast-gradient-sign fooling-images gradient-ascent pytorch pytorch-tutorial
Last synced: 30 Jan 2025
https://github.com/mit-ll-responsible-ai/hydra-zen
Create powerful Hydra applications without the yaml files and boilerplate code.
configurable configuration dataclasses deep-learning dynamic-configuration machine-learning machinelearning pytorch pytorch-lightning reproducibility reproducible-science scalable yaml-configuration
Last synced: 01 Feb 2025
https://github.com/ybabakhin/kaggle_salt_bes_phalanx
Winning solution for the Kaggle TGS Salt Identification Challenge.
deep-learning image-segmentation keras pseudo-labeling pytorch semantic-segmentation u-net
Last synced: 30 Jan 2025
https://github.com/maum-ai/hififace
Unofficial PyTorch Implementation for HifiFace (https://arxiv.org/abs/2106.09965)
face-swapping pytorch pytorch-lightning wandb
Last synced: 05 Feb 2025
https://github.com/RLE-Foundation/RLeXplore
RLeXplore provides stable baselines of exploration methods in reinforcement learning, such as intrinsic curiosity module (ICM), random network distillation (RND) and rewarding impact-driven exploration (RIDE).
baselines efficient-algorithm exploration-strategy gym machine-learning pybullet pytorch reinforcement-learning robotics toolbox
Last synced: 04 Dec 2024
https://github.com/stared/thinking-in-tensors-writing-in-pytorch
Thinking in tensors, writing in PyTorch (a hands-on deep learning intro)
deep-learning pytorch pytorch-tutorial tensor
Last synced: 04 Feb 2025
https://github.com/francescosaveriozuppichini/pytorch-deep-learning-template
A Pytorch Computer Vision template to quick start your next project! 🚀🚀
computer-vision deep-learning python pytorch
Last synced: 02 Feb 2025
https://github.com/square/pysurvival
Open source package for Survival Analysis modeling
deep-learning machine-learning numpy python pytorch survival-analysis
Last synced: 12 Nov 2024
https://github.com/dvl-tum/mot_neural_solver
Official PyTorch implementation of "Learning a Neural Solver for Multiple Object Tracking" (CVPR 2020 Oral).
graph-neural-networks multi-object-tracking pytorch
Last synced: 28 Oct 2024
https://github.com/beomi/infinitransformer
Unofficial PyTorch/🤗Transformers(Gemma/Llama3) implementation of Leave No Context Behind: Efficient Infinite Context Transformers with Infini-attention
gemma huggingface infinitransformer llama llama3 pytorch transformers
Last synced: 04 Feb 2025
https://github.com/antixk/pytorch-model-compare
Compare neural networks by their feature similarity
cka deep-learning feature-extraction imagenet neural-networks pip pytorch torch-cka transformers
Last synced: 04 Feb 2025
https://github.com/megvii-research/MOTRv2
[CVPR2023] MOTRv2: Bootstrapping End-to-End Multi-Object Tracking by Pretrained Object Detectors
end-to-end multi-object-tracking pytorch transformer
Last synced: 28 Oct 2024
https://github.com/uber-research/DeepPruner
DeepPruner: Learning Efficient Stereo Matching via Differentiable PatchMatch (ICCV 2019)
iccv2019 patchmatch pytorch real-time stereo-matching stereo-vision
Last synced: 31 Oct 2024
https://github.com/Megvii-CSG/MegReader
A research project for text detection and recognition using PyTorch 1.2.
ctc deep-learning ocr pytorch text-detection text-detection-recognition text-recognition
Last synced: 04 Nov 2024
https://github.com/imoonlab/meshnet
MeshNet: Mesh Neural Network for 3D Shape Representation (AAAI 2019)
3d-shapes deep-learning mesh pytorch
Last synced: 03 Feb 2025
https://github.com/hzwer/cvpr2023-dmvfn
CVPR2023 (highlight) - A Dynamic Multi-Scale Voxel Flow Network for Video Prediction
computer-vision cvpr2023 deep-learning pytorch
Last synced: 03 Feb 2025
https://github.com/davidmascharka/tbd-nets
PyTorch implementation of "Transparency by Design: Closing the Gap Between Performance and Interpretability in Visual Reasoning"
deep-learning machine-learning neural-networks pytorch visual-question-answering visualization vqa
Last synced: 06 Feb 2025
https://github.com/sony/model_optimization
Model Compression Toolkit (MCT) is an open source project for neural network model optimization under efficient, constrained hardware. This project provides researchers, developers, and engineers advanced quantization and compression tools for deploying state-of-the-art neural networks.
deep-learning deep-neural-networks edge-ai machine-learning network-compression network-quantization neural-network optimizer ptq pytorch qat quantization tensorflow
Last synced: 01 Feb 2025
https://github.com/atgambardella/pytorch-es
Evolution Strategies in PyTorch
machinelearning python pytorch
Last synced: 02 Nov 2024
https://github.com/nicola-decao/s-vae-pytorch
Pytorch implementation of Hyperspherical Variational Auto-Encoders
deep-learning hyperspherical-vae machine-learning manifold-learning pytorch vae variational-autoencoder von-mises-fisher
Last synced: 23 Nov 2024
https://github.com/asyml/vision-transformer-pytorch
Pytorch version of Vision Transformer (ViT) with pretrained models. This is part of CASL (https://casl-project.github.io/) and ASYML project.
Last synced: 30 Jan 2025
https://github.com/AI4Finance-Foundation/FinRL_Podracer
Cloud-native Financial Reinforcement Learning
algorithmic-trading deep-reinforcement-learning machine-learning-algorithms openai-gym pytorch quant quantitative-finance
Last synced: 28 Nov 2024
https://github.com/nashory/delf-pytorch
PyTorch Implementation of "Large-Scale Image Retrieval with Attentive Deep Local Features"
image-retrieval local-features pytorch
Last synced: 30 Jan 2025
https://github.com/mir-group/allegro
Allegro is an open-source code for building highly scalable and accurate equivariant deep learning interatomic potentials
atomistic-simulations computational-chemistry deep-learning drug-discovery force-fields interatomic-potentials machine-learning materials-science molecular-dynamics pytorch
Last synced: 19 Nov 2024
https://github.com/IBM/aihwkit
IBM Analog Hardware Acceleration Kit
ai analog-devices cuda neural-networks pytorch
Last synced: 17 Nov 2024
https://github.com/justusschock/shapenet
PyTorch implementation of "Super-Realtime Facial Landmark Detection and Shape Fitting by Deep Regression of Shape Model Parameters" predicting facial landmarks with up to 400 FPS
deep-learning delira facial-keypoints facial-keypoints-cnn facial-landmarks landmark-detection pytorch shape shape-models shape-prediction shape-predictor
Last synced: 06 Feb 2025
https://github.com/bshall/hubert
HuBERT content encoders for: A Comparison of Discrete and Soft Speech Units for Improved Voice Conversion
pytorch representation-learning speech voice-conversion
Last synced: 03 Feb 2025
https://github.com/lambdalabsml/distributed-training-guide
Best practices & guides on how to write distributed pytorch training code
cluster cuda deepspeed distributed-training fsdp gpu gpu-cluster kuberentes lambdalabs mpi nccl pytorch sharding slurm
Last synced: 03 Feb 2025
https://github.com/vballoli/nfnets-pytorch
NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
adaptive-gradient-clipping deepmind image-classification nfnets paper pytorch sgd sota
Last synced: 28 Oct 2024
https://github.com/ekzhang/fastseg
📸 PyTorch implementation of MobileNetV3 for real-time semantic segmentation, with pretrained weights & state-of-the-art performance
aspp cityscapes computer-vision deep-learning deeplabv3 edge-computing efficientnet kitti-dataset mapillary-vistas-dataset mobilenetv3 pytorch semantic-segmentation
Last synced: 31 Jan 2025
https://github.com/apple/ml-gmpi
[ECCV 2022, Oral Presentation] Official PyTorch implementation of GMPI
3d eccv2022 generative-adversarial-network generative-model multiplane-images neural-rendering pytorch
Last synced: 05 Feb 2025
https://github.com/microsoft/nn-meter
A DNN inference latency prediction toolkit for accurately modeling and predicting the latency on diverse edge devices.
deep-learning deep-neural-networks edge-ai edge-computing efficient-model inference latency machine-learning neural-architecture-search onnx-models python pytorch tensorflow-models
Last synced: 31 Jan 2025
https://github.com/inisis/brocolli
Everything in Torch Fx
caffe onnx pytorch quantization
Last synced: 01 Feb 2025
https://github.com/pytorch/torchx
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.
airflow aws-batch components deep-learning distributed-training kubernetes machine-learning pipelines python pytorch ray slurm
Last synced: 30 Jan 2025
https://github.com/xiuyu-li/q-diffusion
[ICCV 2023] Q-Diffusion: Quantizing Diffusion Models.
ddim diffusion-models model-compression post-training-quantization pytorch quantization stable-diffusion
Last synced: 05 Feb 2025
https://github.com/dmmiller612/sparktorch
Train and run Pytorch models on Apache Spark.
apache-spark deep-learning distributed-computing inference pipelines pytorch sparktorch
Last synced: 01 Feb 2025
https://github.com/Pointcept/PointTransformerV2
[NeurIPS'22] An official PyTorch implementation of PTv2.
Last synced: 28 Oct 2024
https://github.com/zhangmozhe/Deep-Exemplar-based-Video-Colorization
The source code of CVPR 2019 paper "Deep Exemplar-based Video Colorization".
colorization computer-vision deep-learning gan generative-adversarial-network image-generation image-processing old-photo pytorch
Last synced: 28 Nov 2024
https://github.com/taishan1994/pytorch_triple_extraction
基于pytorch的中文三元组提取(命名实体识别+关系抽取)
bert named-entity-recognition pytorch relation-extraction triple-extraction
Last synced: 06 Feb 2025
https://github.com/kevinnan-teen/intelligent-traffic-based-on-cv
本项目是基于计算机视觉的端到端交通路口智能监控系统.采用的设计架构由SRS流媒体服务器、GPU服务器、Local客户端三部分组成.可将远端视频流通过rtmp协议传输到流媒体服务器,然后经过目标检测等一些列算法对视频进行分析,最后在本地客户端查看分析结果.项目主要用Python实现,流媒体服务器采用开源的SRS实时视频服务器搭建,GPU服务器使用YOLO模型实现道路目标如人、车、交通灯等物体的识别,同时还支持并发连接.客户端基于PyQt5进行开发.
Last synced: 06 Feb 2025
https://github.com/Algolzw/image-restoration-sde
Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.
deblurring dehazing denoising deraining diffusion-models image-restoration inpainting pytorch score-based stochastic-differential-equations super-resolution
Last synced: 30 Oct 2024
https://github.com/ZhangGongjie/Meta-DETR
[T-PAMI 2022] Meta-DETR for Few-Shot Object Detection: Official PyTorch Implementation
computer-vision few-shot-object-detection meta-learning object-detection pytorch
Last synced: 15 Oct 2024
https://github.com/lucasalegre/morl-baselines
Multi-Objective Reinforcement Learning algorithms implementations.
gym gymnasium mo-gymnasium morl multi-objective multi-objective-rl pytorch reinforcement-learning rl rl-algorithms
Last synced: 01 Feb 2025
https://github.com/chokkan/mlnote
機械学習帳
matplotlib numpy python pytorch scikit-learn
Last synced: 11 Nov 2024
https://github.com/mjhydri/BeatNet
BeatNet is state-of-the-art (Real-Time) and Offline joint music beat, downbeat, tempo, and meter tracking system using CRNN and particle filtering. (ISMIR 2021's paper implementation).
beat-tracking beatnet crnn-network dnn-beat-tracking downbeat-tracking joint-model meter-detection online-time-signature-detection particle-filtering pytorch real-time real-time-beat-tracker real-time-downbeat-tracker real-time-tempo state-of-the-art
Last synced: 22 Nov 2024
https://github.com/wpeebles/g.pt
Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"
deep-learning diffusion-models pytorch transformers
Last synced: 30 Jan 2025
https://github.com/golsun/DialogRPT
EMNLP 2020: "Dialogue Response Ranking Training with Large-Scale Human Feedback Data"
conversational-ai dataset dialog dialog-datasets dialogpt gpt-2 human-feedback-data pretrained-models pytorch transformers
Last synced: 30 Oct 2024
https://github.com/lukecavabarrett/pna
Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric
dgl graph-machine-learning graph-neural-networks pytorch pytorch-geometric
Last synced: 28 Nov 2024
https://github.com/v-iashin/SpecVQGAN
Source code for "Taming Visually Guided Sound Generation" (Oral at the BMVC 2021)
audio audio-generation bmvc evaluation-metrics gan melgan multi-modal pytorch transformer vas vggsound video video-features video-understanding vqvae
Last synced: 06 Nov 2024
https://github.com/SamsungLabs/adaptis
[ICCV19] AdaptIS: Adaptive Instance Selection Network, https://arxiv.org/abs/1909.07829
adaptis cityscapes instance-segmentation mapillary-vistas-dataset ms-coco mxnet panoptic-segmentation pytorch
Last synced: 08 Nov 2024
https://github.com/Junjue-Wang/LoveDA
[NeurIPS 2021] LoveDA: A Remote Sensing Land-Cover Dataset for Domain Adaptive Semantic Segmentation
domain-adaptation pytorch remote-sensing semantic-segmentation
Last synced: 15 Nov 2024
https://github.com/amirbar/detreg
Official implementation of the CVPR 2022 paper "DETReg: Unsupervised Pretraining with Region Priors for Object Detection".
deep-learning object-detection pytorch unsupervised-learning
Last synced: 05 Feb 2025
https://github.com/xplip/pixel
Research code for pixel-based encoders of language (PIXEL)
deep-learning deep-neural-networks language-model machine-learning nlp pytorch
Last synced: 05 Feb 2025
https://github.com/bshall/zerospeech
VQ-VAE for Acoustic Unit Discovery and Voice Conversion
acoustic-features pytorch speech-synthesis voice-conversion vq-vae zerospeech
Last synced: 06 Feb 2025
https://github.com/jacobgil/pytorch-explain-black-box
PyTorch implementation of Interpretable Explanations of Black Boxes by Meaningful Perturbation
blackbox deep-learning deeplearning pytorch visualization
Last synced: 06 Feb 2025
https://github.com/myungsub/cain
Source code for AAAI 2020 paper "Channel Attention Is All You Need for Video Frame Interpolation"
aaai2020 channel-attention dataset deep-convolutional-networks frame-interpolation pytorch video-frame-interpolation
Last synced: 06 Feb 2025
https://github.com/bethgelab/model-vs-human
Benchmark your model on out-of-distribution datasets with carefully collected human comparison data (NeurIPS 2021 Oral)
benchmark pytorch robustness tensorflow toolbox
Last synced: 15 Nov 2024
https://github.com/domainadaptation/salad
A toolbox for domain adaptation and semi-supervised learning. Contributions welcome.
deep-learning domain-adaptation machine-learning pytorch
Last synced: 14 Nov 2024
https://github.com/leopard-ai/betty
Betty: an automatic differentiation library for generalized meta-learning and multilevel optimization
artificial-intelligence autodiff automatic-differentiation bilevel-optimization deep-learning hyperparameter-optimization implicit-differentiation machine-learning meta-learning multilevel-optimization neural-architecture-search pytorch reinforcement-learning
Last synced: 10 Nov 2024
https://github.com/BIGBALLON/PyTorch-CPP
PyTorch C++ inference with LibTorch
cpp demo imagenet inference libtorch opencv pytorch pytorch-cpp
Last synced: 27 Oct 2024
https://github.com/iMoonLab/MeshNet
MeshNet: Mesh Neural Network for 3D Shape Representation (AAAI 2019)
3d-shapes deep-learning mesh pytorch
Last synced: 28 Oct 2024
https://github.com/vita-group/sinnerf
[ECCV 2022] "SinNeRF: Training Neural Radiance Fields on Complex Scenes from a Single Image", Dejia Xu, Yifan Jiang, Peihao Wang, Zhiwen Fan, Humphrey Shi, Zhangyang Wang
gan nerf neural-radiance-fields pytorch pytorch-lightning
Last synced: 30 Jan 2025
https://github.com/AntixK/PyTorch-Model-Compare
Compare neural networks by their feature similarity
cka deep-learning feature-extraction imagenet neural-networks pip pytorch torch-cka transformers
Last synced: 15 Nov 2024
https://github.com/szymonmaszke/torchdatasets
PyTorch dataset extended with map, cache etc. (tensorflow.data like)
cache concatenate dataset disk filter library map pytorch tensorflow tf-data torch
Last synced: 05 Feb 2025
https://github.com/hancyran/RepSurf
[CVPR 2022 Oral] Official implementation for "Surface Representation for Point Clouds"
cvpr2022 modelnet40 point-cloud pytorch s3dis scannet scanobjectnn sunrgbd
Last synced: 28 Oct 2024
https://github.com/rfeinman/pytorch-minimize
Newton and Quasi-Newton optimization with PyTorch
minimization newton-method optimization pytorch quasi-newton-method
Last synced: 03 Feb 2025
https://github.com/keonlee9420/PortaSpeech
PyTorch Implementation of PortaSpeech: Portable and High-Quality Generative Text-to-Speech
deep-neural-networks fastspeech generative-model hifi-gan high-quality mel-gan neural-tts non-ar non-autoregressive normalizing-flows portable-tts pytorch speech-synthesis text-to-speech tts vae
Last synced: 27 Nov 2024
https://github.com/nimarb/pytorch_influence_functions
This is a PyTorch reimplementation of Influence Functions from the ICML2017 best paper: Understanding Black-box Predictions via Influence Functions by Pang Wei Koh and Percy Liang.
deep-learning influence-functions pytorch pytorch-implementation
Last synced: 01 Feb 2025
https://github.com/ecs-vlc/FMix
Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'
augmentation pytorch pytorch-lightning tensorflow torchbearer
Last synced: 13 Nov 2024
https://github.com/vinhkhuc/PyTorch-Mini-Tutorials
Minimal tutorials for PyTorch
deep-learning machine-learning python pytorch
Last synced: 02 Nov 2024
https://github.com/frgfm/holocron
PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Other additions: AdEMAMix
ademamix computer-vision cspdarknet53 darknet deep-learning object-detection pytorch resnet rexnet tridentnet unet-image-segmentation yolo yolov4
Last synced: 01 Feb 2025
https://github.com/bfortuner/pytorch-kaggle-starter
Pytorch starter kit for Kaggle competitions
deep-learning jupyter-notebook kaggle kaggle-competition machine-learning python3 pytorch pytorch-tutorial
Last synced: 02 Feb 2025
https://github.com/yuanyuanli85/Fast_Human_Pose_Estimation_Pytorch
Pytorch Code for CVPR2019 paper "Fast Human Pose Estimation" https://arxiv.org/abs/1811.05419
knowledge-distillation pose-estimation pytorch stacked-hourglass-networks
Last synced: 05 Nov 2024
https://github.com/filaPro/oneformer3d
[CVPR2024] OneFormer3D: One Transformer for Unified Point Cloud Segmentation
3d-instance-segmentation 3d-object-detection 3d-panoptic-segmentation 3d-semantic-segmentation mmdetection3d pytorch s3dis scannet scannet200
Last synced: 30 Dec 2024
https://github.com/bknyaz/graph_nn
Graph Classification with Graph Convolutional Networks in PyTorch (NeurIPS 2018 Workshop)
gcn graph graph-classification graph-convolutional-networks graph-unet multigraph pooling pytorch
Last synced: 12 Nov 2024
https://github.com/benedekrozemberczki/attentionwalk
A PyTorch Implementation of "Watch Your Step: Learning Node Embeddings via Graph Attention" (NeurIPS 2018).
attention deep-learning deepwalk graph-attention graph-neural-networks graph-representation-learning implicit-factorization machine-learning matrix-factorization neurips neurips-2018 nips node2vec pytorch sklearn structural-attention tensorflow torch walklet word2vec
Last synced: 06 Feb 2025
https://github.com/liuruoze/mini-alphastar
(JAIR'2022) A mini-scale reproduction code of the AlphaStar program. Note: the original AlphaStar is the AI proposed by DeepMind to play StarCraft II. JAIR = Journal of Artificial Intelligence Research.
deep-learning deep-neural-networks deep-reinforcement-learning gaming imitation-learning mini-alphastar multi-agent-systems pytorch reinforcement-learning sc2replay self-playing-bot starcraft-ii-bot starcraft2 starcraft2-ai supervised-learning
Last synced: 06 Feb 2025