下载数据集
将train.py中下载数据集代码部分的参数改为true，运行后可自动下载数据集
训练
运行train.py进行训练
测试
运行predict.py进行训练结果的预测
-
Notifications
You must be signed in to change notification settings - Fork 0
dtbdtbdtb/pytorch_test
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
这是一个pytorch demo，用以学习如何使用pytorch搭建神经网络并通过训练数据集来实现一个分类任务。
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published