PyTorch Examples

Explore community-driven projects and learn PyTorch.

Image Classification with ResNet

A comprehensive example demonstrating image classification using the ResNet architecture. Includes data loading, model training, and evaluation.

Computer Vision CNN ResNet ImageNet

Natural Language Processing: Text Generation

Build a recurrent neural network (RNN) based model for generating text sequences, trained on a custom corpus. Explore various sampling strategies.

NLP RNN Text Generation LSTMs

Reinforcement Learning: CartPole Solver

Implement a Deep Q-Network (DQN) agent to solve the classic CartPole problem from OpenAI Gym. Understand policy learning and reward maximization.

Reinforcement Learning DQN Gym AI

Object Detection with YOLOv5

A practical guide to setting up and training a YOLOv5 model for real-time object detection on custom datasets.

Computer Vision Object Detection YOLO Real-time

Time Series Forecasting with LSTMs

Utilize Long Short-Term Memory (LSTM) networks for forecasting future values in time series data, such as stock prices or weather patterns.

Time Series Forecasting LSTM Data Science

Generative Adversarial Networks (GANs) for Image Generation

Explore the principles of GANs by building a model that generates new, realistic images, such as faces or handwritten digits.

GANs Generative Models Image Generation Deep Learning