Open weitianli opened 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
When I run the Cifar10 I met such a problem. (Python 3.6, Pytorch 0.4.0)
I wrote init.py to fix it.