# PyTorch学习

**Repository Path**: i7606/pytorch-learning

## Basic Information

- **Project Name**: PyTorch学习
- **Description**: 学习PyTorch和相关案例的仓库
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 0
- **Created**: 2026-01-27
- **Last Updated**: 2026-05-18

## Categories & Tags

**Categories**: Uncategorized

**Tags**: Python

## README

# PyTorch学习

## 一、介绍

学习PyTorch和相关案例的仓库

## 二、软件架构

1、MNIST - 手写数字识别，编写了一个窗口用于测试效果

![mnist window](./images/mnist.png "mnist window")

2、CNN-MNIST -手写数字识别的卷积神经网络版本，识别率更高

![cnn-mnist window](./images/cnn-mnist.png "cnn-mnist window")

3、Gpt-learn —— GPT生成 ~~唐诗~~ 宋词风格诗句（可能宋词语料更多导致整体风格更偏向于宋词），原先的模型效果不好，经过检查后发现少用了一个归一化层导致的问题，已修复并重新训练模型，现在效果还可以。

~~Gpt-learn -GPT生成唐诗风格语句，能看出来风格与唐诗相似，输出文字没有具体含义（因为仅训练25万步，所以效果不太好，如果训练到50万步效果将更好）~~

![gpt-song-poetry](./images/gpt-song-poetry.png "gpt song poetry")

## 三、使用方式

1. 进入对应文件夹
2. 控制台执行 `pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128`
3. 控制台执行 `pip3 install`
4. 控制台执行 `python ./main.py` 运行