AttributeError: module 'models' has no attribute 'VGG'

[weitianli]

When I run the Cifar10 I met such a problem. (Python 3.6, Pytorch 0.4.0)

Namespace(arch='VGG16', batch_size=128, cuda=True, epochs=300, evaluate=False, log_interval=100, lr=0.1, lr_epochs=100, momentum=0.9, no_cuda=False, pretrained=None, seed=1, test_batch_size=100, weight_decay=1e-05)
Files already downloaded and verified
Traceback (most recent call last):
  File "main.py", line 341, in <module>
    model_ori = models.VGG(name)
AttributeError: module 'models' has no attribute 'VGG'

I wrote init.py to fix it.

from .VGG import VGG
from .Bin_VGG import Bin_VGG_test
from .Bin_VGG import Bin_VGG_train