Update lenet.md (d2l-ai#457)

panhongjin · Dec 25, 2018 · b7cf0d0 · b7cf0d0
1 parent 7df2501
commit b7cf0d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chapter_convolutional-neural-networks/lenet.md b/chapter_convolutional-neural-networks/lenet.md
@@ -63,15 +63,15 @@ train_iter, test_iter = gb.load_data_fashion_mnist(batch_size=batch_size)
 因为卷积神经网络计算比多层感知机要复杂，建议使用GPU来加速计算。我们尝试在`gpu(0)`上创建NDArray，如果成功则使用`gpu(0)`，否则仍然使用CPU。
 
 ```{.python .input}
-def try_gpu4():  # 本函数已保存在 gluonbook 包中方便以后使用。
+def try_gpu():  # 本函数已保存在 gluonbook 包中方便以后使用。
     try:
         ctx = mx.gpu()
         _ = nd.zeros((1,), ctx=ctx)
     except mx.base.MXNetError:
         ctx = mx.cpu()
     return ctx
 
-ctx = try_gpu4()
+ctx = try_gpu()
 ctx
 ```