Skip to content

Commit 983c3af

Browse files
authored
Update README.md
1 parent b29b43e commit 983c3af

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

README.md

+6-1
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,14 @@
11
# pytorch_train
2+
## update
3+
4+
vgg增加一个全连接层,并加入dropout层,数据增强,batchsize设为100,top1准确率可以达到90.34%
25

36
---
47
Lenet的准确率约为53%
58

6-
修改后的VGG16准确率约为76%(训练时batchsize设为100,迭代20次),后续可做实验再进行优化(不加BN层的话VGG16的网络不会收敛)不同batchsize对正确率会有影响,设batchsize为4时准确率可到80%
9+
修改后的VGG16准确率约为76%(训练时batchsize设为100,迭代20次),后续可做实验再进行优化(不加BN层的话VGG16的网络不会收敛)不同batchsize对正确率会有影响,设batchsize为4时准确率可到80%
10+
11+
712

813
resnet18的准确率也为约76%(训练时batchsize设为100,同样batchsize训练90次,并加入学习率衰减机制,准确率也可以达到85%),同样不同batchsize对正确率会有影响,设batchsize为4时(训练20次)准确率可到84%,更换为Adam优化算法后(并训练90次,加入学习率衰减机制),可以将准确率提升到85.3%
914

0 commit comments

Comments
 (0)