Skip to content

Commit a81cc5e

Browse files
committed
debug cifar-demo.py
1 parent 7f2fcd6 commit a81cc5e

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,6 @@ To train cifar-net on cifar dataset, do as follow:
1111
1. cd code
1212
2. python cifar_demo.py
1313

14-
This is a brief CNN tutorial (from [Jianxin Wu](http://cs.nju.edu.cn/wujx/)'s personal website):
14+
This is a brief CNN tutorial (from [Jianxin Wu](http://cs.nju.edu.cn/wujx/)'s homepage):
1515

1616
http://cs.nju.edu.cn/_upload/tpl/00/ed/237/template237/paper/CNN.pdf

code/cifar-demo.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,7 @@
2222
train_data, valid_data, train_labels, valid_labels = load_cifar()
2323

2424
cnn = initial_cifar()
25-
lr = np.ones(20) * 0
26-
.01
25+
lr = np.ones(20) * 0.001
2726
cnn.train(train_data, train_labels, lr, epoch=20, batch_size=32)
2827

2928
res = cnn.predict(valid_data)

0 commit comments

Comments
 (0)