This commit is contained in:
mohankumarSriram 2018-11-28 23:02:10 +05:30 коммит произвёл Tianqi Chen
Родитель 55599b937b
Коммит 10ce048c79
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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())
######################################################################