Batch Normalization
Training Deep Networks
Why do we need batch normalization layers? Let us review some practical challenges that arise when training neural networks.
First, the way data are preprocessed often dramatically influences the final result. Recall the example of using a multilayer perceptron to predict house prices. When working with real data, our fir ...
Posted on Fri, 08 May 2026 10:39:23 +0000 by Gorf
Cat vs Dog Recognition with LeNet and PyTorch
01 Cat vs Dog Recognition
Introduction: Manually building LeNet for cat vs dog recognition.
Reference: https://mtyjkh.blog.csdn.net/article/details/121263237
Code: 01-cat-dog (github.com)
Note: Beginners are advised to practice typing all the code, as it serves as a template. Regardless, you should be able to type it fluently (know the steps, t ...
Posted on Thu, 07 May 2026 14:18:47 +0000 by vbracknell
Saving and Loading Models in PyTorch Networks
Synthetic Training Data Generation
import torch
import torch.nn.functional as F
import matplotlib.pyplot as plt
import numpy as np
# Create synthetic dataset
train_x = torch.linspace(-1, 1, 100).view(-1, 1)
train_y = train_x ** 2 + 0.2 * torch.rand(train_x.size())
# Visualize input-output distribution
plt.scatter(train_x.numpy(), train_y.nump ...
Posted on Thu, 07 May 2026 11:30:23 +0000 by ArmanIc
Troubleshooting Common Issues in Docker GPU Training Environments
Checking Your System Configuration
To begin, verify your system's configuration with these commands:
Check the kernel version used by your NVIDIA driver:
cat /proc/driver/nvidia/version
View installed NVIDIA packages:
cat /var/log/dpkg.log | grep nvidia
List all NVIDIA drivers installed:
sudo dpkg --list | grep nvidia-*
Problem 1: NVML Initial ...
Posted on Thu, 07 May 2026 08:30:34 +0000 by harsha