45 skills found · Page 1 of 2
NVIDIA / ApexA PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
openvinotoolkit / NncfNeural Network Compression Framework for enhanced OpenVINO™ inference
yg211 / Bert NliA Natural Language Inference (NLI) model based on Transformers (BERT and ALBERT)
voldemortX / DST CBCImplementation of our Pattern Recognition paper "DMT: Dynamic Mutual Training for Semi-Supervised Learning"
dronefreak / VisDrone Dataset Python ToolkitModern PyTorch toolkit for the VisDrone aerial object detection dataset with production-ready training pipelines, real-time inference, and format converters. Features state-of-the-art models (Faster R-CNN, FCOS, RetinaNet), mixed precision training, rich progress tracking, and optimizations for small object detection in drone imagery.
mrusci / Training Mixed Precision Quantized NetworksThis repository containts the pytorch scripts to train mixed-precision networks for microcontroller deployment, based on the memory contraints of the target device.
mcarilli / Mixed Precision ReferencesPersonal collection of references for high performance mixed precision training.
ayminovitch / Fine Tune CodebaseFine-Tune Codebase Model 🚀 A scalable and efficient tool for fine-tuning large language models (LLMs) on codebases. Supports LoRA, mixed precision training, quantization, and interactive testing. Perfect for adapting models to your specific codebase, whether it's small or large-scale.
ByungKwanLee / Super Fast Adversarial TrainingOfficial PyTorch Implementation Code for Developing Super Fast Adversarial Training with Distributed Data Parallel, Channel Last Memory Format, Mixed Precision Training + Adversarial Attack, Faster Adversarial Training, and Fast Forward Computer Vision (FFCV).
tspeterkim / Mixed Precision From ScratchMixed precision training from scratch with Tensors and CUDA
lliai / EMQ Series[ICCV-2023] EMQ: Evolving Training-free Proxies for Automated Mixed Precision Quantization
TigerJeffX / Transformers In PytorchNo description available
UbiquitousLearning / Mandheling DSP TrainingThe open-source project for "Mandheling: Mixed-Precision On-Device DNN Training with DSP Offloading"[MobiCom'2022]
AberHu / ImageNet TrainingPytorch ImageNet training codes with various tricks, lr schedulers, distributed training, mixed precision training, DALI dataloader etc.
kbhujbal / LungGuardianAI Pneumonia Detection From Chest Xrays With Transfer LearningState-of-the-Art Pneumonia detection from chest X-rays system using EfficientNetV2 + FPN + Faster R-CNN. Features Focal Loss, Weighted Box Fusion, Mosaic Augmentation & StratifiedGroupKFold. Built for RSNA Pneumonia Detection Challenge. Achieves competitive mAP with mixed-precision training.
khcs / Fp16 Demo TfExamples for mixed-precision training for utilizing TensorCores in NVIDIA Volta GPUs
Data-Science-in-Mechanical-Engineering / Mixed Precision For JAXA repository for Mixed Precision Training in JAX based on Equinox.
johnolafenwa / TorchFusion UtilsA pytorch helper library for Mixed Precision Training, Initialization, Metrics and More Utilities to simplify training of deep learning models
mratsim / Humpback Whale IdentificationKaggle Humpback whale identification: 2xGPU Data augmentation + FP16 mixed precision training
kentaroy47 / Pytorch Cifar10 Fp16Let's train CIFAR 10 Pytorch with Half-Precision!