Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
Yakraii authored Jul 4, 2024
1 parent 5bdae26 commit 83108c1
Show file tree
Hide file tree
Showing 17 changed files with 2,706,028 additions and 2 deletions.
38 changes: 36 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,2 +1,36 @@
# pytorch-cifar10
www.kaggle.com/c/cifar-10
# ResNet on Cifar10

[kaggle-cifar10](http://www.kaggle.com/c/cifar-10)

## 环境

- python3.9
- PyTorch version: 2.2.0+cpu



## 运行

#### 1.数据集

按照说明文档中的数据集结构(使用d2l的方法)处理好数据放置于'./data'中。

#### 2.训练模型

运行'cifar10-train.ipynb'中代码即可,其中包含resnet18(修改), resnet56。

#### 3.测试模型

导入model中保存的模型,使用'cifar10-train.ipynb'最后部分的预测生成'submisson.csv',上传kaggle评分。



## 效果

#### resnet56

测试准确率达到92.46%

#### resnet18(修改)

测试准确率达到89.94%
Loading

0 comments on commit 83108c1

Please sign in to comment.