Commit 10ce048c by mohankumarSriram Committed by Tianqi Chen

fixing nnvm tutorial typo (#2188)

parent 55599b93
......@@ -50,7 +50,7 @@ data_shape = (batch_size,) + image_shape
out_shape = (batch_size, num_class)
net, params = nnvm.testing.resnet.get_workload(
layers=18, batch_size=batch_size, image_shape=image_shape)
num_layers=18, batch_size=batch_size, image_shape=image_shape)
print(net.debug_str())
######################################################################
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment