# 基于pytorch的CIFAR10的简单分类

**Repository Path**: chenkun_code/pycharm_cifar10

## Basic Information

- **Project Name**: 基于pytorch的CIFAR10的简单分类
- **Description**: 通过对CIFAR10的简单分类，发现在模型训练中可能存在的问题。
- **Primary Language**: Python
- **License**: MulanPSL-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-02-08
- **Last Updated**: 2024-03-18

## Categories & Tags

**Categories**: Uncategorized

**Tags**: Python, PyTorch, Deep-learning

## README

# 基于pytorch的CIFAR10的简单分类

## 介绍
通过对CIFAR10的简单分类，发现在模型训练中可能存在的问题。


可选参数:


  --batch_size          样本个数，默认是64


  --epoch               训练次数，默认是10


  --lr                  学习率，默认为0.001


  --cuda                是否使用cuda训练
  


## 示例:

### 训练


python "your_file.py"  --batch_size=64  --lr=0.001 --epoch=20 --cuda

### 损失值绘图

tensorboard --logdir=“your_data_path”