serend1p1ty / core-pytorch-utils
Yet another PyTorch Trainer and some core components for deep learning.
☆214Updated 11 months ago
Alternatives and similar repositories for core-pytorch-utils:
Users that are interested in core-pytorch-utils are comparing it to the libraries listed below
- The pure and clear PyTorch Distributed Training Framework.☆276Updated last year
- 这里是改进了pytorch的DataParallel, 用来平衡第一个GPU的显存使用量☆232Updated 4 years ago
- PyTorch Project Specification.☆679Updated 3 years ago
- pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用☆94Updated last year
- Simple tutorials on Pytorch DDP training☆275Updated 2 years ago
- 一款便捷的抢占显卡脚本☆314Updated 2 months ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆91Updated 2 years ago
- 蜻蜓点论文 Think不Clear, 论文解读视频上传B站, youtube, 西瓜视频(同步到抖音)☆242Updated last year
- 📊 A simple command-line utility for querying and monitoring GPU status☆90Updated 2 years ago
- ☆62Updated 3 years ago
- Some tricks of pytorch...☆1,182Updated 9 months ago
- How to use wandb?☆628Updated last year
- Pytorch Lightning入门中文教程,转载请注明来源。(当初是写着玩的,建议看完MNIST这个例子再上手)☆207Updated 4 years ago
- A bag of tricks to speed up your deep learning process☆160Updated 11 months ago
- Implementation of Swin Transformer with Pytorch☆111Updated 3 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆188Updated 2 years ago
- PASSL包含 SimCLR,MoCo v1/v2,BYOL,CLIP,PixPro,simsiam, SwAV, BEiT,MAE 等图像自监督算法以及 Vision Transformer,DEiT,Swin Transformer,CvT,T2T-ViT,MLP-…☆282Updated last year
- [ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?☆331Updated 2 years ago
- (WIP) TorchUtils is a pytorch library with several useful tools and training tricks.☆87Updated 2 years ago
- 整理 pytorch 单机多 GPU 训练方法与原理☆806Updated 3 years ago
- ☆425Updated 2 years ago
- ☆64Updated 2 years ago
- Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"☆361Updated last year
- Implementation of some unbalanced loss like focal_loss, dice_loss, DSC Loss, GHM Loss et.al☆260Updated 2 years ago
- To be the world's best PyTorch project template.☆503Updated 2 years ago
- The implementation of mixup and mainfold mixup method with standard models(PreActRes, WideRes, Dense) in Cifar10, Cifar100 and SVHN datas…☆44Updated 3 years ago
- 多模态 MM +Chat 合集☆249Updated last month
- 全中文注释.(The loss function of retinanet based on pytorch).(You can use it on one-stage detection task or classifical task, to solve data …☆475Updated 9 months ago
- Official MegEngine implementation of RepLKNet☆275Updated 2 years ago
- a collection of visualization function☆414Updated 3 years ago