← all repositories

lartpang/PyTorchTricks

A collection of practical PyTorch optimization tricks covering model design, inference acceleration, quantization, and memory optimization.

1.2k stars ML FrameworksLearning
PyTorchTricks
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository documents various techniques for improving PyTorch model training and inference. It covers topics like model design patterns, convolution-BN fusion for faster inference, model compression and quantization methods, memory-efficient training strategies such as mixed precision with apex, and accelerated data loading using tar format with IterableDataset. Content is organized by category with links to reference papers and resources.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.