# pytorch复现经典论文系列

**Repository Path**: lin-binghua/DL_Torch

## Basic Information

- **Project Name**: pytorch复现经典论文系列
- **Description**: 一些经典深度学习复现pytorch代码及工程代码
- **Primary Language**: Python
- **License**: MulanPSL-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2025-07-17
- **Last Updated**: 2025-07-17

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# pytorch复现经典论文系列

#### 介绍
基于pytorch手写复现深度学习历史上的一些经典论文，并记录一些实验相关,简洁版本。

#### 软件架构
pytorch 1.12.0+cu102
python3.8

#### 使用说明

Alexnet:
pytorch 数据准备及训练测试脚本

0. split_data.py  将数据集划分为训练集测试集。 

[数据集url](http://www.cvmart.net/dataSets/detail/526?channel_id=op10&utm_source=cvmartmp&utm_campaign=datasets&utm_medium=article)

```
python split_data.py
```

原始数据格式：


```
├── flowers
│   ├── daisy
│   ├── dandelion
│   ├── rose
│   ├── sunflower
│   └── tulip
```



划分成训练测试后


```
.
├── test
│   ├── daisy
│   ├── dandelion
│   ├── rose
│   ├── sunflower
│   └── tulip
└── train
    ├── daisy
    ├── dandelion
    ├── rose
    ├── sunflower
    └── tulip
```


1.  对应模型名称文件夹下的model.py为模型的pytorch定义。
2.  对应模型名称文件夹下的train.py为模型的训练代码。 python train.py
3.  对应模型名称文件夹下的infer.py为模型的测试代码。 python infer.py