cooooorn / Pytorch-XNOR-Net

XNOR-Net, with binary gemm and binary conv2d kernels, support both CPU and GPU.
BSD 3-Clause "New" or "Revised" License
82 stars 21 forks source link

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

Open weitianli opened 6 years ago

weitianli commented 6 years ago

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