Implementing Automatic Mixed Precision Training in PyTorch
PyTorch's Automatic Mixed Precision (AMP) feature allows efficient training by combining FP32 and FP16 precision operations. This technique reduces memory usage and accelerates computation while maintaining model accuracy.
Understanding Mixed Precision
Deep learning models traditionally use 32-bit floating point (FP32) for all operations. Mixed ...
Posted on Wed, 01 Jul 2026 16:52:27 +0000 by Miker
PyTorch Embedding Layer Mechanics and Linear Layer Differences
Lookup Table Mechanics
In neural networks for sequence processing, the nn.Embedding module functions as a searchable dictionary. It translates discrete integer identifiers into continuous high-dimensional vectors. Rather than requiring sparse one-hot representations as inputs, this layer dircetly accepts integer indices to retrieve their corres ...
Posted on Sun, 28 Jun 2026 18:02:56 +0000 by Sander
Getting Started with Torch-Pruning for Structured Model Pruning
Torch-Pruning is a PyTorch library designed for structured model pruning. It leverages DepGraph (Dependency Graph) to automatically identify and manage inter-layer dependencies, ensuring that pruning operations preserve model integrity during forward passes.
Installation
Clone the repository and install in development mode:
git clone https:/ ...
Posted on Sat, 27 Jun 2026 16:05:36 +0000 by chipev
Essential Steps for Getting Started with Deep Learning
Deep learning, a specialized subset of machine learning, utilizes artificial neural networks with multiple layers to model complex patterns in data. This foundational technology powers advancements in computer vision, speech synthesis, and language understanding.
Core Components of Neural Networks
Neural networks consist of interconnected layer ...
Posted on Sat, 20 Jun 2026 17:14:54 +0000 by knetcozd
Deploying RKNN Models: Evaluation and Inference Testing
Differences Between Loading Native and RKNN-Converted Models
Models developed in frameworks like PyTorch, TensorFlow, or ONNX must be converted into the proprietary RKNN format to leverage Rockchip’s NPU aceleration. The RKNN format is optimized for Rockchip’s neural processing units, enabling efficient execution on embedded platforms such as t ...
Posted on Mon, 15 Jun 2026 18:27:05 +0000 by dwest
Comprehensive Guide to Installing PyTorch on Windows, Jetson Nano, and Ubuntu
This guide walks through the setup of PyTorch across three common environments: Windows (with CUDA), NVIDIA Jetson Nano (JetPack 4.6), and Ubuntu Linux. It covers driver installation, CUDA Toolkit, cuDNN, and final verification.
Windows Installation
1.1 Verify or Install NVIDIA Driver
The NVIDIA driver acts as the communication bridge betwee ...
Posted on Mon, 15 Jun 2026 15:59:59 +0000 by Johannes80
Advanced Custom Layer Development in PyTorch: Implementation Patterns
PyTorch Extension Framework Fundamentals
PyTorch's architecture enables deep customization through two primary extension points. Understanding their distinct capabilities is essential for implementing novel neural network components.
nn.Module: Parameterized Component Foundation
The nn.Module class serves as the cornerstone for trainable compon ...
Posted on Sat, 13 Jun 2026 16:11:25 +0000 by arjuna
Fine-Tuning ResNet for Hotdog Image Classification Using Transfer Learning
Steps
Below we introduce fine‑tuning, a common technique in transfer learning. As illustrated in the following diagram, fine‑tuning consists of four steps.
Pre‑train a neural network model (the source model) on a source dataset, e.g., ImageNet.
Create a new neural network (the target model). It replicates all model design and parameters from t ...
Posted on Fri, 12 Jun 2026 17:03:14 +0000 by aniket_dj
Surface Crack Detection with CNN on Kaggle
1. Dataset Acquisition
Concrete surface cracks are a primary defect in civil structures. Building inspection is performed to evaluate stiffness and tensile strangth. Crack detection plays a vital role in building health assessment.
The dataset contains images of various concrete surfaces with and without cracks. The image data is divided into t ...
Posted on Thu, 11 Jun 2026 17:53:30 +0000 by Jedi Legend
Pipeline Parallelism in Large-Scale AI Model Training
Training large-scale neural networks often exceeds the computational and memory capacity of a single device, neecssitating distributed training strategies. Among these, model parallelism (MP) plays a crucial role by partitioning the model itself across multiple devices. Within MP, pipeline parallelism (PP) stands out as an effective technique t ...
Posted on Thu, 11 Jun 2026 16:37:39 +0000 by smonkcaptain