We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 713e29b commit 80de0a3Copy full SHA for 80de0a3
utee/selector.py
@@ -74,7 +74,7 @@ def vgg16(cuda=True, model_root=None):
74
def vgg16_bn(cuda=True, model_root=None):
75
print("Building vgg16_bn parameters")
76
from imagenet import vgg
77
- m = vgg.vgg19_bn(model_root)
+ m = vgg.vgg16_bn(model_root)
78
if cuda:
79
m = m.cuda()
80
return m, dataset.get, True
0 commit comments