提交 dff01de9 编写于 作者: M Megvii Engine Team

fix(advanced): fix typo

GitOrigin-RevId: db034ca012aed20966d265f3a35baff676d5c909
上级 8bf7c48d
......@@ -64,7 +64,7 @@
batch_data = np.random.randn(batch_size, 3, 224, 224).astype(np.float32)
batch_label = np.random.randint(1000, size=(batch_size,)).astype(np.int32)
data.set_value(batch_data)
label.set_label(batch_label)
label.set_value(batch_label)
optimizer.zero_grad()
train_func(data, label, net=resnet, optimizer=optimizer)
optimizer.step()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册