Skip to content

cyd03/CNN_CIFAR10_Classify

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

环境: pytorch python 3.11

基于卷积网络实现的CIFAR-10的图像分类问题 main.py中存放主要的训练代码和测试代码 test.py是对img中自定义数据集的测试 model.py中为卷积网络模型的建立 data将下载CIFAR-10的测试集和训练集 logs_4用于tensorboard的可视化 在运行main.py文件前先删去logs_4文件夹,是logs_4能够更新。 其中tt_99.pth是训练100次后保存的参数,即已经训练好的模型。

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages