search

skhadem / 3D-BoundingBox

PyTorch implementation for 3D Bounding Box Estimation Using Deep Learning and Geometry
MIT License
435 stars 96 forks source link

Error model = Model(features=my_vgg.features, bins=2).cuda() #8

Closed mheriyanto closed 4 years ago

mheriyanto commented 4 years ago

When I run this code: my_vgg = vgg.vgg19_bn(pretrained=True) model = Model(features=my_vgg.features, bins=2).cuda() I get notice below: model = Model(features=my_vgg.features, bins=2).cuda() TypeError: 'module' object is not callable

mheriyanto commented 4 years ago

I get solution. I change model = Model(features=my_vgg.features, bins=2).cuda() to model = Model.Model(features=my_vgg.features, bins=2).cuda()