Skip to content

Commit 80de0a3

Browse files
author
chenxi
committed
fix issue44
1 parent 713e29b commit 80de0a3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

utee/selector.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def vgg16(cuda=True, model_root=None):
7474
def vgg16_bn(cuda=True, model_root=None):
7575
print("Building vgg16_bn parameters")
7676
from imagenet import vgg
77-
m = vgg.vgg19_bn(model_root)
77+
m = vgg.vgg16_bn(model_root)
7878
if cuda:
7979
m = m.cuda()
8080
return m, dataset.get, True

0 commit comments

Comments
 (0)