Skip to content

Commit e45bd30

Browse files
authored
Merge pull request oreilly-japan#8 from TanUkkii007/fix-nameerror-in-train_convnet.py
Fix NameError
2 parents ebea9b1 + 0ba1f2f commit e45bd30

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ch07/train_convnet.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,12 +14,14 @@
1414
#x_train, t_train = x_train[:5000], t_train[:5000]
1515
#x_test, t_test = x_test[:1000], t_test[:1000]
1616

17+
max_epochs = 20
18+
1719
network = SimpleConvNet(input_dim=(1,28,28),
1820
conv_param = {'filter_num': 30, 'filter_size': 5, 'pad': 0, 'stride': 1},
1921
hidden_size=100, output_size=10, weight_init_std=0.01)
2022

2123
trainer = Trainer(network, x_train, t_train, x_test, t_test,
22-
epochs=20, mini_batch_size=100,
24+
epochs=max_epochs, mini_batch_size=100,
2325
optimizer='Adam', optimizer_param={'lr': 0.001},
2426
evaluate_sample_num_per_epoch=1000)
2527
trainer.train()

0 commit comments

Comments
 (0)